diff --git a/.config/nvim/init.lua b/.config/nvim/init.lua
index 5da97ac..1109b14 100644
--- a/.config/nvim/init.lua
+++ b/.config/nvim/init.lua
@@ -32,6 +32,9 @@ require("nvim-tree").setup({
     dotfiles = false,
     custom = { "^.git$" },
   },
+  update_focused_file = {
+    enable = true,
+  },
 })
 
 -- mason
diff --git a/.config/nvim/lua/func.lua b/.config/nvim/lua/func.lua
index c1dc278..a868be9 100644
--- a/.config/nvim/lua/func.lua
+++ b/.config/nvim/lua/func.lua
@@ -15,10 +15,10 @@ vim.cmd [[
 vim.cmd [[
   augroup lang_puppet
       autocmd!
-      autocmd FileType puppet nnoremap <buffer> <Leader>gg :call OpenPuppetClassOrTemplate(tab)<CR>
-      autocmd FileType puppet nnoremap <buffer> <Leader>gt :call OpenPuppetClassOrTemplate(tab)<CR>
-      autocmd FileType puppet nnoremap <buffer> <Leader>gh :call OpenPuppetClassOrTemplate(horizontal)<CR>
-      autocmd FileType puppet nnoremap <buffer> <Leader>gv :call OpenPuppetClassOrTemplate(vertical)<CR>
+      autocmd FileType puppet nnoremap <buffer> <Leader>gg :call OpenPuppetClassOrTemplate('tab')<CR>
+      autocmd FileType puppet nnoremap <buffer> <Leader>gt :call OpenPuppetClassOrTemplate('tab')<CR>
+      autocmd FileType puppet nnoremap <buffer> <Leader>gh :call OpenPuppetClassOrTemplate('horizontal')<CR>
+      autocmd FileType puppet nnoremap <buffer> <Leader>gv :call OpenPuppetClassOrTemplate('vertical')<CR>
       autocmd FileType puppet nnoremap <buffer> <Leader>t :call OpenPuppetTestMode()<CR>
       autocmd BufNewFile site/roles/manifests/**.pp call ApplyPuppetTemplate()
       autocmd BufNewFile site/profiles/manifests/**.pp call ApplyPuppetTemplate()
diff --git a/.config/nvim/lua/func/puppet.lua b/.config/nvim/lua/func/puppet.lua
index 3983ffd..34ef9a5 100644
--- a/.config/nvim/lua/func/puppet.lua
+++ b/.config/nvim/lua/func/puppet.lua
@@ -116,19 +116,19 @@ endfunction
 
 -- open puppet resource
 vim.cmd([[
-function! OpenPuppetClassOrTemplate()
+function! OpenPuppetClassOrTemplate(layout)
   " Get the line under the cursor
   let line = getline(".")
 
   " Check if it's a template (erb or epp)
   if line =~ 'profiles\/.*\.\(erb\|epp\)'
-    call OpenPuppetTemplate()
+    call OpenPuppetTemplate(a:layout)
     return
   endif
 
   " Check if it's an included class (profiles:: or roles::)
   if line =~ '\(profiles\|roles\)::'
-    call OpenPuppetProfileOrRole()
+    call OpenPuppetProfileOrRole(a:layout)
     return
   endif