diff --git a/tests/unit/general/test_retry_scheduler.cpp b/tests/unit/general/test_retry_scheduler.cpp
index 5f5fea0..296c855 100644
--- a/tests/unit/general/test_retry_scheduler.cpp
+++ b/tests/unit/general/test_retry_scheduler.cpp
@@ -132,11 +132,10 @@ TEST_F_S(Schedule, ImmediateExecution) {
   },
     1ms);
 
-  while (!first_call_scheduler_thread_id || !second_call_scheduler_thread_id) {
+  while (m_impl.isScheduled()) {
     std::this_thread::sleep_for(1ms);
   }
 
-  EXPECT_FALSE(m_impl.isScheduled());
   EXPECT_TRUE(first_call_scheduler_thread_id);
   EXPECT_TRUE(second_call_scheduler_thread_id);