add return statement at __enter__