Fix tests

xm-online · Jun 6, 2024 · fdc44e4 · fdc44e4
1 parent 2b50c8b
commit fdc44e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/com/icthh/xm/tmf/ms/activation/config/TaskIterationLepAdditionalContext.java b/src/main/java/com/icthh/xm/tmf/ms/activation/config/TaskIterationLepAdditionalContext.java
@@ -35,7 +35,7 @@ public Optional<Integer> additionalContextValue(BaseLepContext lepContext, LepEn
         LepBaseKey lepBaseKey = lepMethod.getLepBaseKey();
         if ("tasks".equals(lepBaseKey.getGroup()) && "Task".equals(lepBaseKey.getBaseKey())) {
             SagaEvent sagaEvent = lepMethod.getParameter("sagaEvent", SagaEvent.class);
-            return Optional.ofNullable(sagaEvent.getIteration());
+            return Optional.ofNullable(sagaEvent).map(SagaEvent::getIteration);
         } else {
             return Optional.empty();
         }