Fix Connection context manager