diff --git a/lib/test_wifi.py b/lib/test_wifi.py
index 880c089..94a64ba 100644
--- a/lib/test_wifi.py
+++ b/lib/test_wifi.py
@@ -16,8 +16,8 @@ class TestWifi(unittest.TestCase):
         Pin(Pin.PWM_LCD_BLIGHT).off()
 
     def test_connect(self):
-        wifi.connect()
-        self.assertTrue(wifi.is_connected(show_wait_message=True))
+        wifi.connect(show_wait_message=True)
+        self.assertTrue(wifi.is_connected())
 
 if __name__ == '__main__':
     unittest.main()