diff --git a/cleanlab_studio/internal/api/api.py b/cleanlab_studio/internal/api/api.py
index 1cea9f0c..b251d506 100644
--- a/cleanlab_studio/internal/api/api.py
+++ b/cleanlab_studio/internal/api/api.py
@@ -534,7 +534,7 @@ def deploy_model(api_key: str, cleanset_id: str, model_name: str) -> str:
     return model_id
 
 
-def get_deployment_status(api_key: str, model_id: str) -> JSONDict:
+def get_deployment_status(api_key: str, model_id: str) -> str:
     """Gets status of model deployment."""
     check_uuid_well_formed(model_id, "model ID")
     res = requests.get(
diff --git a/cleanlab_studio/studio/studio.py b/cleanlab_studio/studio/studio.py
index 125ed6c3..1416e1b8 100644
--- a/cleanlab_studio/studio/studio.py
+++ b/cleanlab_studio/studio/studio.py
@@ -328,7 +328,7 @@ def deploy_model(self, cleanset_id: str, model_name: str) -> str:
         """
         return api.deploy_model(self._api_key, cleanset_id, model_name)
 
-    def wait_until_model_ready(self, model_id: str, timeout: Optional[float] = None) -> str:
+    def wait_until_model_ready(self, model_id: str, timeout: Optional[float] = None) -> None:
         """Blocks until a model is ready or the timeout is reached.
 
         Args: