diff --git a/src/aiia/__init__.py b/src/aiia/__init__.py
index 6d716f8..838c765 100644
--- a/src/aiia/__init__.py
+++ b/src/aiia/__init__.py
@@ -1,4 +1,4 @@
-from .model.Model import AIIABase, AIIABaseShared, AIIAchunked, AIIAExpert, AIIAmoe, AIIArecursive, AIIASparseMoe
+from .model.Model import AIIABase, AIIABaseShared, AIIAchunked, AIIAmoe, AIIASparseMoe, AIIArecursive
 from .model.config import AIIAConfig
 from .data.DataLoader import DataLoader
 from .pretrain.pretrainer import Pretrainer, ProjectionHead
diff --git a/src/aiia/model/__init__.py b/src/aiia/model/__init__.py
index c0ad6e7..a45512a 100644
--- a/src/aiia/model/__init__.py
+++ b/src/aiia/model/__init__.py
@@ -2,10 +2,9 @@ from .Model import (
     AIIABase,
     AIIABaseShared,
     AIIAchunked,
-    AIIAExpert,
     AIIAmoe,
-    AIIArecursive,
-    AIIASparseMoe
+    AIIASparseMoe,
+    AIIArecursive
 )
 from .config import AIIAConfig
 
@@ -13,9 +12,9 @@ __all__ = [
     "AIIABase",
     "AIIABaseShared",
     "AIIAchunked",
-    "AIIAExpert",
     "AIIAmoe",
+    "AIIASparseMoe",
     "AIIArecursive",
     "AIIAConfig",
-    "AIIASparseMoe"
+
 ]
\ No newline at end of file