Needed to return self on __enter__