Fix the "Connection lost" error