diff --git a/lua/nvim-tree/lib.lua b/lua/nvim-tree/lib.lua
index 15fea88e..cbdcdd71 100644
--- a/lua/nvim-tree/lib.lua
+++ b/lua/nvim-tree/lib.lua
@@ -121,12 +121,12 @@ function M.unroll_dir(node)
     git.update_gitignore_map_sync()
     populate(node.entries, node.link_to or node.absolute_path, node)
 
-    if vim.g.nvim_tree_lsp_diagnostics == 1 then
-      diagnostics.update()
-    end
-
     renderer.draw(M.Tree, true)
   end
+
+  if vim.g.nvim_tree_lsp_diagnostics == 1 then
+    diagnostics.update()
+  end
 end
 
 local function refresh_git(node, update_gitignore)