diff --git a/src/great_ai/great_ai/deploy/great_ai.py b/src/great_ai/great_ai/deploy/great_ai.py
index 462e7cd..0f4cde3 100644
--- a/src/great_ai/great_ai/deploy/great_ai.py
+++ b/src/great_ai/great_ai/deploy/great_ai.py
@@ -123,6 +123,7 @@ class GreatAI(Generic[T]):
                 Callable[[Callable[..., T]], GreatAI[T]],
                 partial(
                     GreatAI.create,
+                    version=version,
                     return_raw_result=return_raw_result,
                     disable_rest_api=disable_rest_api,
                     disable_docs=disable_docs,