diff --git a/tests/gateway/test_manage.py b/tests/gateway/test_manage.py
index ac633c9..6b5ae16 100644
--- a/tests/gateway/test_manage.py
+++ b/tests/gateway/test_manage.py
@@ -26,6 +26,7 @@ class GatewayTest(unittest.TestCase):
     def test_run_multiple_errors(self, mock_gateway):
         instance = mock_gateway.return_value
         instance.run.side_effect = [BrokenPipeError(), socket.error(), KeyboardInterrupt()]
+        instance.disconnect.side_effect = [True, True, OSError()]
 
         run("user_2")