diff --git a/modules/state_flow/state_flow.module b/modules/state_flow/state_flow.module
index 26aad4f..7a672d6 100644
--- a/modules/state_flow/state_flow.module
+++ b/modules/state_flow/state_flow.module
@@ -418,6 +418,9 @@ function state_flow_prevent_live_revision($node) {
     // redirect the user to the revision saved. This hijacks the redirection by
     // drupal_goto().
     $_GET['destination'] = 'node/' . $node->nid . '/revisions/' . $node->vid . '/view';
+
+    // Save the vid for the draft revision back to the node in case other modules need it.
+    $node->new_draft_vid = $node->vid;
   }
 }