diff --git a/spring-statemachine-core/src/test/java/org/springframework/statemachine/AbstractStateMachineTests.java b/spring-statemachine-core/src/test/java/org/springframework/statemachine/AbstractStateMachineTests.java
index 64b642c3..2595263a 100644
--- a/spring-statemachine-core/src/test/java/org/springframework/statemachine/AbstractStateMachineTests.java
+++ b/spring-statemachine-core/src/test/java/org/springframework/statemachine/AbstractStateMachineTests.java
@@ -178,8 +178,8 @@ public abstract class AbstractStateMachineTests {
 			if (message != null) {
 				log.info(this);
 			}
-			onExecuteLatch.countDown();
 			stateContexts.add(context);
+			onExecuteLatch.countDown();
 		}
 	}