diff --git a/unit_test.c b/unit_test.c
index b9cf34a88c309d8783680d7fe3e0a5eefd81399a..13ce3f795d1c30f3e6a423ac0f6e147bfca1f550 100644
--- a/unit_test.c
+++ b/unit_test.c
@@ -414,7 +414,7 @@ static const char *test_server(void) {
   ASSERT(mg_connect(server, "127.0.0.1", atoi(HTTP_PORT),  0, ts1, buf1) == 1);
   ASSERT(mg_connect(server, "127.0.0.1", atoi(HTTP_PORT), 0, ts2, buf2) == 1);
 
-  { int i; for (i = 0; i < 50; i++) mg_poll_server(server, 0); }
+  { int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1); }
   ASSERT(strcmp(buf1, "foo ? 127.0.0.1") == 0);
   ASSERT(strcmp(buf2, "404 ERR: 404") == 0);
 
@@ -498,7 +498,7 @@ static const char *test_mg_connect(void) {
   ASSERT(mg_connect(server, "127.0.0.1", 29, 0, cb3, buf3) == 1);
   ASSERT(mg_connect(server, "127.0.0.1", atoi(HTTP_PORT), 0, cb4, buf4) == 1);
 
-  { int i; for (i = 0; i < 50; i++) mg_poll_server(server, 0); }
+  { int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1); }
 
   ASSERT(strcmp(buf2, "add") == 0);
   ASSERT(strcmp(buf3, "1") == 0);