diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java
index 03cb101a9d..0de61da3bd 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/ForkJoinStateMachineTest.java
@@ -22,7 +22,7 @@ import static org.junit.Assert.assertTrue;
 public class ForkJoinStateMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java
index 950414bb0e..76105368d8 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/HierarchicalStateMachineTest.java
@@ -21,7 +21,7 @@ import static org.junit.Assert.assertEquals;
 public class HierarchicalStateMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java
index 64930162fd..b91e6af5c5 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/JunctionStateMachineTest.java
@@ -19,7 +19,7 @@ import javax.annotation.Resource;
 public class JunctionStateMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java
index b7cbebe145..cd9e58eb8e 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateEnumMachineTest.java
@@ -23,7 +23,7 @@ import static org.junit.Assert.assertTrue;
 public class StateEnumMachineTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<ApplicationReviewStates, ApplicationReviewEvents> stateMachine;
 
     @Before
     public void setUp() {
diff --git a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java
index 8f61d93105..1c1ab58917 100644
--- a/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java
+++ b/spring-state-machine/src/test/java/com/baeldung/spring/statemachine/StateMachineIntegrationTest.java
@@ -22,7 +22,7 @@ import javax.annotation.Resource;
 public class StateMachineIntegrationTest {
 
     @Resource
-    private StateMachine stateMachine;
+    private StateMachine<String, String> stateMachine;
 
     @Before
     public void setUp() {