diff --git a/after/ftplugin/lua.vim b/after/ftplugin/lua.vim
index 73b1a8f..6fdbc66 100644
--- a/after/ftplugin/lua.vim
+++ b/after/ftplugin/lua.vim
@@ -6,3 +6,7 @@ nnoremap <buffer><silent> <F9> :luafile %<CR>
 
 " For delimitMate
 let b:delimitMate_matchpairs = "(:),[:],{:}"
+
+" Use nvim-treesitter for folding
+set foldmethod=expr
+set foldexpr=nvim_treesitter#foldexpr()
diff --git a/after/ftplugin/python.vim b/after/ftplugin/python.vim
index 185aa6e..71e31c0 100644
--- a/after/ftplugin/python.vim
+++ b/after/ftplugin/python.vim
@@ -15,3 +15,7 @@ set expandtab       " expand tab to spaces so that tabs are spaces
 
 " For delimitMate
 let b:delimitMate_matchpairs = "(:),[:],{:}"
+
+" Use nvim-treesitter for folding
+set foldmethod=expr
+set foldexpr=nvim_treesitter#foldexpr()