From ce6a296233107e663750e60d0fded2cd5f932dec Mon Sep 17 00:00:00 2001
From: chylex <contact@chylex.com>
Date: Wed, 27 Mar 2024 13:24:17 +0100
Subject: [PATCH] Fix(VIM-3364): Exception with mapped Generate action

---
 src/main/java/com/maddyhome/idea/vim/helper/IjActionExecutor.kt | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/main/java/com/maddyhome/idea/vim/helper/IjActionExecutor.kt b/src/main/java/com/maddyhome/idea/vim/helper/IjActionExecutor.kt
index 07939be64..07375be25 100644
--- a/src/main/java/com/maddyhome/idea/vim/helper/IjActionExecutor.kt
+++ b/src/main/java/com/maddyhome/idea/vim/helper/IjActionExecutor.kt
@@ -20,6 +20,7 @@ import com.intellij.openapi.actionSystem.ex.ActionUtil
 import com.intellij.openapi.actionSystem.ex.ActionUtil.performDumbAwareWithCallbacks
 import com.intellij.openapi.actionSystem.impl.ProxyShortcutSet
 import com.intellij.openapi.actionSystem.impl.SimpleDataContext
+import com.intellij.openapi.actionSystem.impl.Utils
 import com.intellij.openapi.application.ex.ApplicationManagerEx
 import com.intellij.openapi.command.CommandProcessor
 import com.intellij.openapi.command.UndoConfirmationPolicy
@@ -111,6 +112,7 @@ internal class IjActionExecutor : VimActionExecutor {
       ActionManager.getInstance(),
       0,
     )
+    Utils.initUpdateSession(event)
     // beforeActionPerformedUpdate should be called to update the action. It fixes some rider-specific problems
     //   because rider uses an async update method. See VIM-1819.
     // This method executes inside lastUpdateAndCheckDumb