protect changes

backend/classes/galileo_platform.py

@@ -53,11 +53,15 @@ class GalileoPlatform:
             stage_name=self.config.protect_stage_name,
         )
         if not protect_stage:
-            protect_stage = create_protect_stage(
+            _ = create_protect_stage(
                 project_name=self.config.protect_project_name,
                 name=self.config.protect_stage_name,
                 stage_type=StageType.local
             )
+            protect_stage = get_protect_stage(
+                project_name=self.config.protect_project_name,
+                stage_name=self.config.protect_stage_name,
+            )
         return protect_stage.id
 
     def run_protect(self, input: str, output: str, logger: GalileoLogger | None = None) -> dict[Any, Any]: