diff --git a/src/engine/src/core/workflows/ValueFromService.py b/src/engine/src/core/workflows/ValueFromService.py
index 52f4502c..4c406038 100644
--- a/src/engine/src/core/workflows/ValueFromService.py
+++ b/src/engine/src/core/workflows/ValueFromService.py
@@ -25,10 +25,10 @@ def get_task_output_value_by_id(self, task_id, _id):
         return output
     
     def get_env_value_by_key(self, key):
-        output = self._env_repo.get_by_key(key)
+        output = self._env_repo.get_value_by_key(key)
         return output
     
     def get_arg_value_by_key(self, key):
-        output = self._arg_repo.get_by_key(key)
+        output = self._arg_repo.get_value_by_key(key)
         return output
         
\ No newline at end of file