diff --git a/src/great_ai/great_ai/__init__.py b/src/great_ai/great_ai/__init__.py
index f2014ca..04e3e65 100644
--- a/src/great_ai/great_ai/__init__.py
+++ b/src/great_ai/great_ai/__init__.py
@@ -6,7 +6,7 @@ from .exceptions import (
     WrongDecoratorOrderError,
 )
 from .models import save_model, use_model
-from .output_models import (
+from .output_views import (
     ClassificationOutput,
     MultiLabelClassificationOutput,
     RegressionOutput,
diff --git a/src/great_ai/great_ai/models/__init__.py b/src/great_ai/great_ai/models/__init__.py
index bf988ae..ec4cda1 100644
--- a/src/great_ai/great_ai/models/__init__.py
+++ b/src/great_ai/great_ai/models/__init__.py
@@ -1,2 +1,2 @@
 from .save_model import save_model
-from .use_model import use_model
+from .use_model import model_versions, use_model