context managers - fix