diff --git a/src/Controller/ActionController.php b/src/Controller/ActionController.php
index 87276d6..9048876 100644
--- a/src/Controller/ActionController.php
+++ b/src/Controller/ActionController.php
@@ -751,7 +751,7 @@ class ActionController extends ControllerBase {
       try {
         $event = new EntityUpdateByKordiamEvent($node, $data, $kordiam_revision, 'create_by_kordiam');
         \Drupal::service('event_dispatcher')->dispatch($event, EntityUpdateByKordiamEvent::EVENT_NAME);
-        
+        $node = $event->getNode();
         if (!$event->shouldPreventRevisionSave()) {
           $node = ModuleSettings::kordiamRevisionSet($node, $kordiam_revision);
           // Bad updating Kordiam revision.
@@ -969,7 +969,7 @@ class ActionController extends ControllerBase {
     try {
       $event = new EntityUpdateByKordiamEvent($entity, $data, $kordiam_revision, 'update_by_kordiam');
       \Drupal::service('event_dispatcher')->dispatch($event, EntityUpdateByKordiamEvent::EVENT_NAME);
-      
+      $entity = $event->getNode();
       // Only update Kordiam revision if not prevented by event subscribers.
       if (!$event->shouldPreventRevisionSave()) {
         // Updating Kordiam revision.
diff --git a/src/Event/EntityUpdateByKordiamEvent.php b/src/Event/EntityUpdateByKordiamEvent.php
index 69eec4e..91fa7cf 100644
--- a/src/Event/EntityUpdateByKordiamEvent.php
+++ b/src/Event/EntityUpdateByKordiamEvent.php
@@ -67,6 +67,13 @@ class EntityUpdateByKordiamEvent extends Event {
     return $this->node;
   }
 
+  /**
+   * Sets the node.
+   */
+  public function setNode(EntityInterface $node): void {
+    $this->node = $node;
+  }
+
   /**
    * Gets incoming story data.
    */