diff --git a/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java b/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java
index eb53942..7204327 100644
--- a/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java
+++ b/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java
@@ -8,6 +8,7 @@ import java.util.List;
 public class StateMachineTransitionConfigurationMethodFinder {
     private final CompilationUnit cu;
     private final ConfigureMethodVisitor visitor = new ConfigureMethodVisitor();
+    private static final List<String> stateMachineConfigurationAdapterSuperclasses = List.of("EnumStateMachineConfigurerAdapter", "StateMachineConfigurerAdapter");
 
     public StateMachineTransitionConfigurationMethodFinder(String source) {
         ASTParser parser = ASTParser.newParser(AST.getJLSLatest());
@@ -47,9 +48,9 @@ public class StateMachineTransitionConfigurationMethodFinder {
 
             if (superclass.isParameterizedType()) {
                 ParameterizedType pt = (ParameterizedType) superclass;
-                return pt.getType().toString().equals("StateMachineConfigurerAdapter");
+                return stateMachineConfigurationAdapterSuperclasses.contains(pt.getType().toString());
             }
-            return superclass.toString().equals("StateMachineConfigurerAdapter");
+            return stateMachineConfigurationAdapterSuperclasses.contains(superclass.toString());
         }
 
         private boolean isConfigureMethod(MethodDeclaration method) {