diff --git a/lua/valeth/plugins/ufo.lua b/lua/valeth/plugins/ufo.lua
index 66c0e18..332d422 100644
--- a/lua/valeth/plugins/ufo.lua
+++ b/lua/valeth/plugins/ufo.lua
@@ -24,7 +24,9 @@ spec.config = function()
     })
 
     ufo.setup({
-        close_fold_kinds = {"comment", "imports"},
+        close_fold_kinds_for_ft = {
+            default = { "comment", "imports" },
+        },
     })
 end