diff --git a/tests/test_ptdm.py b/tests/test_ptdm.py
index abc02dbec..34dbd9f03 100644
--- a/tests/test_ptdm.py
+++ b/tests/test_ptdm.py
@@ -43,14 +43,19 @@ def callback_with_args():
         )
         client.start_listening()
 
+        assert callback_without_args.counter == 0
+        assert callback_with_args.counter == 0
+
+        # Emit message that emits events without arguments
         self.socket_mock.recv_string.return_value = f"{Message.PUB_LOW_BATTERY_WARNING}"
-        wait_until(lambda: callback_without_args.counter > 0, timeout=3)
+        wait_until(lambda: callback_without_args.counter > 0, timeout=20)
         assert callback_with_args.counter == 0
 
+        # Emit message that emits events with 1 argument
         self.socket_mock.recv_string.return_value = (
             f"{Message.PUB_BRIGHTNESS_CHANGED}|1"
         )
-        wait_until(lambda: callback_with_args.counter > 0, timeout=3)
+        wait_until(lambda: callback_with_args.counter > 0, timeout=20)
 
         client.stop_listening()