diff --git a/geti_sdk/rest_clients/model_client.py b/geti_sdk/rest_clients/model_client.py
index dbf4afb5..1f2d3473 100644
--- a/geti_sdk/rest_clients/model_client.py
+++ b/geti_sdk/rest_clients/model_client.py
@@ -622,7 +622,7 @@ def optimize_model(self, model: Model, optimization_type: str = "pot") -> Job:
                 f"Invalid optimization type specified: `{optimization_type}`. Valid "
                 f"options are: {valid_optimization_types}"
             )
-        optimize_model_url = model.base_url + "/optimize"
+        optimize_model_url = model.base_url + ":optimize"
         payload = {
             "enable_nncf_optimization": optimization_type == "nncf",
             "enable_pot_optimization": optimization_type == "pot",
diff --git a/geti_sdk/rest_clients/training_client.py b/geti_sdk/rest_clients/training_client.py
index db1095d0..c233071d 100644
--- a/geti_sdk/rest_clients/training_client.py
+++ b/geti_sdk/rest_clients/training_client.py
@@ -236,7 +236,7 @@ def train_task(
                 self.monitor_jobs(jobs=task_training_jobs, timeout=timeout)
 
         response = self.session.get_rest_response(
-            url=f"{self.base_url}/train", method="POST", data=data
+            url=f"{self.base_url}:train", method="POST", data=data
         )
 
         job_id = response["job_id"]