diff --git a/lua/easytables/window.lua b/lua/easytables/window.lua
index 53b5eee..0a9f2f1 100644
--- a/lua/easytables/window.lua
+++ b/lua/easytables/window.lua
@@ -173,6 +173,7 @@ function M:register_listeners()
     vim.api.nvim_create_autocmd(
         { "QuitPre" },
         {
+            buffer = self.prompt_buffer,
             callback = function()
                 self:close()
             end