diff --git a/init.lua b/init.lua
index 06b23a2b..a6aa858e 100644
--- a/init.lua
+++ b/init.lua
@@ -672,7 +672,8 @@ require('lazy').setup({
         end,
       })
 
-      vim.api.nvim_create_autocmd('InsertLeave', {
+      vim.api.nvim_create_autocmd('ModeChanged', {
+        pattern = 'i:*',
         callback = function()
           set_virtual_text(false)
         end,