diff --git a/src/packages/tree-view/lib/dialog.coffee b/src/packages/tree-view/lib/dialog.coffee
index e63425a8d..ddb71181a 100644
--- a/src/packages/tree-view/lib/dialog.coffee
+++ b/src/packages/tree-view/lib/dialog.coffee
@@ -32,7 +32,7 @@ class Dialog extends View
 
   close: ->
     @remove()
-    $('#root-view').focus()
+    rootView.focus()
 
   cancel: ->
     @remove()