diff --git a/tests/client/test_client_ssh.c b/tests/client/test_client_ssh.c
index 06f935b..6d211cf 100644
--- a/tests/client/test_client_ssh.c
+++ b/tests/client/test_client_ssh.c
@@ -524,6 +524,7 @@
     session = nc_connect_ssh("127.0.0.1", 8080, NULL);
     assert_non_null(session);
 
+    will_return(__wrap_ssh_channel_poll_timeout, 0);
     nc_session_free(session, NULL);
 }
 
@@ -574,6 +575,7 @@
     assert_non_null(session);
 
     /* disconnect */
+    will_return(__wrap_ssh_channel_poll_timeout, 0);
     nc_session_free(session, NULL);
 }
 
@@ -613,6 +615,7 @@
     assert_non_null(session);
 
     /* disconnect */
+    will_return(__wrap_ssh_channel_poll_timeout, 0);
     nc_session_free(session, NULL);
 }
 
@@ -625,6 +628,7 @@
     errno = ECONNREFUSED;
     will_return(__wrap_connect, -1);
     will_return(__wrap_ssh_is_connected, 0);
+    will_return(__wrap_ssh_channel_poll_timeout, 0);
 
     session = nc_connect_ssh("127.0.0.1", 8080, NULL);
     assert_null(session);
@@ -654,6 +658,7 @@
     will_return(__wrap_ssh_channel_open_session, 0);
     will_return(__wrap_ssh_channel_request_subsystem, 0);
     will_return(__wrap_nc_handshake_io, 4);
+    will_return(__wrap_ssh_channel_poll_timeout, 0);
 
     session = nc_connect_ssh("127.0.0.1", 8080, NULL);
     assert_null(session);