diff --git a/ai_toolkit/models/__init__.py b/ai_toolkit/models/__init__.py
index 775da11..a6cf4ba 100644
--- a/ai_toolkit/models/__init__.py
+++ b/ai_toolkit/models/__init__.py
@@ -24,9 +24,9 @@ def get_model_initializer(model_name: str) -> type[nn.Module]:
 
 __all__ = (
     "BasicCNN",
-    "DenseNet",
     "BasicLSTM",
-    "MaskRCNN",
     "BasicRNN",
+    "DenseNet",
+    "MaskRCNN",
     "get_model_initializer",
 )