diff --git a/pydra/engine/core.py b/pydra/engine/core.py
index a523c24525..d0081e3ace 100644
--- a/pydra/engine/core.py
+++ b/pydra/engine/core.py
@@ -380,7 +380,7 @@ def can_resume(self):
         return self._can_resume
 
     @abc.abstractmethod
-    def _run_task(self):
+    def _run_task(self, environment=None):
         pass
 
     @property
@@ -1329,7 +1329,7 @@ async def _run(self, submitter=None, rerun=False, **kwargs):
         self._check_for_hash_changes()
         return result
 
-    async def _run_task(self, submitter, rerun=False):
+    async def _run_task(self, submitter, rerun=False, environment=None):
         if not submitter:
             raise Exception("Submitter should already be set.")
         for nd in self.graph.nodes: