diff --git a/agent/src/main/scala/za/co/absa/atum/agent/AtumAgent.scala b/agent/src/main/scala/za/co/absa/atum/agent/AtumAgent.scala
index 4c29517e8..8644acde9 100644
--- a/agent/src/main/scala/za/co/absa/atum/agent/AtumAgent.scala
+++ b/agent/src/main/scala/za/co/absa/atum/agent/AtumAgent.scala
@@ -57,7 +57,7 @@ class AtumAgent private[agent] () {
     val newPartitions: AtumPartitions = parentAtumContext.atumPartitions ++ subPartitions
 
     val newPartitionsDTO = AtumPartitions.toSeqPartitionDTO(newPartitions)
-    val maybeParentPartitionsDTO = Some(AtumPartitions.toSeqPartitionDTO(parentAtumContext.atumPartitions))
+    val parentPartitionsDTO = Some(AtumPartitions.toSeqPartitionDTO(parentAtumContext.atumPartitions))
     val partitioningDTO = PartitioningDTO(newPartitionsDTO, maybeParentPartitionsDTO)
 
     val atumContextDTO = dispatcher.getOrCreateAtumContext(partitioningDTO)