Include __enter__ in shim