diff --git a/docs/mapping.rst b/docs/mapping.rst
index 64c3ecf6a..098cdc96a 100644
--- a/docs/mapping.rst
+++ b/docs/mapping.rst
@@ -210,17 +210,25 @@ In order to make this work, you need to configure your editor as show below:
 
     .. code-block:: lua
 
-        vim.api.nvim_create_autocmd({ "VimEnter", "VimResume" }, {
+        vim.api.nvim_create_autocmd({ "VimEnter", "VimResume", "UIEnter" }, {
             group = vim.api.nvim_create_augroup("KittySetVarVimEnter", { clear = true }),
             callback = function()
-                io.stdout:write("\x1b]1337;SetUserVar=in_editor=MQo\007")
+                if vim.api.nvim_ui_send then
+                    vim.api.nvim_ui_send("\x1b]1337;SetUserVar=in_editor=MQo\007")
+                else
+                    io.stdout:write("\x1b]1337;SetUserVar=in_editor=MQo\007")
+                end
             end,
         })
 
         vim.api.nvim_create_autocmd({ "VimLeave", "VimSuspend" }, {
             group = vim.api.nvim_create_augroup("KittyUnsetVarVimLeave", { clear = true }),
             callback = function()
-                io.stdout:write("\x1b]1337;SetUserVar=in_editor\007")
+                if vim.api.nvim_ui_send then
+                    vim.api.nvim_ui_send("\x1b]1337;SetUserVar=in_editor=MQo\007")
+                else
+                    io.stdout:write("\x1b]1337;SetUserVar=in_editor\007")
+                end
             end,
         })