diff --git a/python/knext/knext/client/marklang/schema_ml.py b/python/knext/knext/client/marklang/schema_ml.py
index 25c180ae..5761e6cc 100644
--- a/python/knext/knext/client/marklang/schema_ml.py
+++ b/python/knext/knext/client/marklang/schema_ml.py
@@ -310,7 +310,7 @@ class SPGSchemaMarkLang:
             spg_type = EntityType(
                 name=f"{self.namespace}.{type_name}", name_zh=type_name_zh
             )
-            if type_class == "EventType":
+            if parent_spg_type.spg_type_enum == SpgTypeEnum.Event:
                 spg_type = EventType(
                     name=f"{self.namespace}.{type_name}", name_zh=type_name_zh
                 )