diff --git a/.config/nvim/lua/my/functions.lua b/.config/nvim/lua/my/functions.lua
index 6056637..174a6c8 100644
--- a/.config/nvim/lua/my/functions.lua
+++ b/.config/nvim/lua/my/functions.lua
@@ -8,6 +8,9 @@ function M.map(mode, shortcut, command, description, buffer)
 end
 
 function M.remove_trailing_whitespace()
+    if vim.b.cleanup_file == false then
+        return
+    end
     local curpos = vim.api.nvim_win_get_cursor(0)
     vim.cmd([[keeppatterns %s/\s\+$//e]])
     vim.api.nvim_win_set_cursor(0, curpos)