Let __enter__ return self and test the ctx manager