diff --git a/candy/lua/marshmallow/defaults.lua b/candy/lua/marshmallow/defaults.lua
index 461a166..c4aefed 100644
--- a/candy/lua/marshmallow/defaults.lua
+++ b/candy/lua/marshmallow/defaults.lua
@@ -59,8 +59,8 @@ vim.opt.shiftwidth = tab_width
 vim.opt.softtabstop = tab_width
 
 -- Folds --
-vim.opt.foldmethod = "expr"
-vim.opt.foldexpr = "nvim_treesitter#foldexpr()"
+vim.wo.foldmethod = "expr"
+vim.wo.foldexpr = "v:lua.vim.treesitter.foldexpr()"
 vim.opt.foldenable = false
 
 -- Is this okay??