diff --git a/pom.xml b/pom.xml
index a5ef9e6..e570d85 100755
--- a/pom.xml
+++ b/pom.xml
@@ -4,7 +4,7 @@
 	<artifactId>lm-lib</artifactId>
 	<groupId>de.extio</groupId>
 	<name>lm-lib</name>
-	<version>1.6.16</version>
+	<version>1.6.17</version>
 	
 	<properties>
 		<java.version>21</java.version>
diff --git a/src/main/java/de/extio/lmlib/agent/BaseAgent.java b/src/main/java/de/extio/lmlib/agent/BaseAgent.java
index 6647c20..3e775ac 100644
--- a/src/main/java/de/extio/lmlib/agent/BaseAgent.java
+++ b/src/main/java/de/extio/lmlib/agent/BaseAgent.java
@@ -76,7 +76,10 @@ public interface BaseAgent {
 		context.setNextAgent(null);
 		
 		this.preProcess(context);
-		
+		if (context.getAgentContextUpdateConsumer() != null) {
+			context.getAgentContextUpdateConsumer().accept(context);
+		}
+
 		splits.addAll(this.applyTemplate(context));
 		final var tasks = new ArrayList<CompletableFuture<?>>(splits.size());
 		for (final var split : splits) {