diff --git a/test/unit/Elsa.Activities.UnitTests/Looping/WhileTests.cs b/test/unit/Elsa.Activities.UnitTests/Looping/WhileTests.cs
index d211963c4..8a25be8c0 100644
--- a/test/unit/Elsa.Activities.UnitTests/Looping/WhileTests.cs
+++ b/test/unit/Elsa.Activities.UnitTests/Looping/WhileTests.cs
@@ -68,9 +68,8 @@ public class WhileTests
 
         // Assert
         Assert.Equal(1, evaluationCount); // Condition should be evaluated once
-        var scheduledActivities = context.WorkflowExecutionContext.Scheduler.List().ToList();
-        Assert.Single(scheduledActivities); // Body should be scheduled once
-        Assert.Equal(bodyActivity, scheduledActivities.First().Activity);
+        var hasBodyScheduledActivity = context.HasScheduledActivity(bodyActivity);
+        Assert.True(hasBodyScheduledActivity);
     }
 
     // Helper methods