Test

Jaseci-Labs · Nov 14, 2024 · 7bc89ac · 7bc89ac
1 parent d5f911a
commit 7bc89ac
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/jac/jaclang/plugin/default.py b/jac/jaclang/plugin/default.py
@@ -591,11 +591,8 @@ def make_architype(
         if not hasattr(cls, "_jac_entry_funcs_") or not hasattr(
             cls, "_jac_exit_funcs_"
         ):
-            # Saving the module path and reassign it after creating cls
-            # So the jac modules are part of the correct module
-            cur_module = cls.__module__
-            cls = type(cls.__name__, (cls, arch_base), {})
-            cls.__module__ = cur_module
+            bases = ((arch_base,) + cls.__bases__) if arch_base not in cls.__bases__ else cls.__bases__
+            cls.__bases__ = bases
             cls._jac_entry_funcs_ = on_entry  # type: ignore
             cls._jac_exit_funcs_ = on_exit  # type: ignore
         else: