Quick __enter__ modification