diff --git a/runtime/src/main/java/org/apache/flink/agents/runtime/operator/ActionExecutionOperator.java b/runtime/src/main/java/org/apache/flink/agents/runtime/operator/ActionExecutionOperator.java
index 60b7e329..a11efc0e 100644
--- a/runtime/src/main/java/org/apache/flink/agents/runtime/operator/ActionExecutionOperator.java
+++ b/runtime/src/main/java/org/apache/flink/agents/runtime/operator/ActionExecutionOperator.java
@@ -825,10 +825,6 @@ private ActionTask createActionTask(Object key, Action action, Event event) {
     }
 
     private void createAndSetRunnerContext(ActionTask actionTask, Object key) {
-        if (actionTask.getRunnerContext() != null) {
-            return;
-        }
-
         RunnerContextImpl runnerContext;
         if (actionTask.action.getExec() instanceof JavaFunction) {
             runnerContext = createOrGetRunnerContext(true);