diff --git a/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java b/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java
index cbb64e46..a8c84717 100644
--- a/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java
+++ b/protocol/src/main/java/com/zfoo/protocol/registration/ProtocolAnalysis.java
@@ -292,8 +292,8 @@ public class ProtocolAnalysis {
                     var protocolId = getProtocolIdAndCheckClass(protocolClass);
                     initProtocolClass(protocolId, protocolClass);
                     protocolModuleSet.add(protocolClass);
-                    classModuleDefinitionMap.put(clazz, moduleId);
-                    protocolXmlEnhanceMap.compute(clazz, (key, value) -> Boolean.TRUE.equals(value) || enhance);
+                    classModuleDefinitionMap.put(protocolClass, moduleId);
+                    protocolXmlEnhanceMap.compute(protocolClass, (key, value) -> Boolean.TRUE.equals(value) || enhance);
                 }
             }
         }