diff --git a/src/aiia/model/__init__.py b/src/aiia/model/__init__.py
index f68a42a..c0ad6e7 100644
--- a/src/aiia/model/__init__.py
+++ b/src/aiia/model/__init__.py
@@ -1,21 +1,21 @@
 from .Model import (
-    AIIA,
     AIIABase,
     AIIABaseShared,
     AIIAchunked,
     AIIAExpert,
     AIIAmoe,
-    AIIArecursive
+    AIIArecursive,
+    AIIASparseMoe
 )
 from .config import AIIAConfig
 
 __all__ = [
-    "AIIA",
     "AIIABase",
     "AIIABaseShared",
     "AIIAchunked",
     "AIIAExpert",
     "AIIAmoe",
     "AIIArecursive",
-    "AIIAConfig"
+    "AIIAConfig",
+    "AIIASparseMoe"
 ]
\ No newline at end of file