fix import in _socket2.py