diff --git a/app/assets/javascripts/dispatcher.js b/app/assets/javascripts/dispatcher.js
index 8d805fc0c1a..7a6165c7714 100644
--- a/app/assets/javascripts/dispatcher.js
+++ b/app/assets/javascripts/dispatcher.js
@@ -9,7 +9,6 @@
 /* global MilestoneSelect */
 /* global Commit */
 /* global NotificationsForm */
-/* global TreeView */
 /* global NotificationsDropdown */
 /* global GroupAvatar */
 /* global LineHighlighter */
@@ -52,6 +51,7 @@ import UsersSelect from './users_select';
 import RefSelectDropdown from './ref_select_dropdown';
 import GfmAutoComplete from './gfm_auto_complete';
 import ShortcutsBlob from './shortcuts_blob';
+import TreeView from './tree';
 import UsagePing from './usage_ping';
 import UsernameValidator from './username_validator';
 import VersionCheckImage from './version_check_image';
diff --git a/app/assets/javascripts/main.js b/app/assets/javascripts/main.js
index 8a90aaf9336..246ab65d54f 100644
--- a/app/assets/javascripts/main.js
+++ b/app/assets/javascripts/main.js
@@ -153,7 +153,6 @@ import './subscription_select';
 import './syntax_highlight';
 import './task_list';
 import './todos';
-import './tree';
 import './user';
 
 // eslint-disable-next-line global-require, import/no-commonjs
diff --git a/app/assets/javascripts/tree.js b/app/assets/javascripts/tree.js
index 1f1fc155fcd..7777ed1c3dc 100644
--- a/app/assets/javascripts/tree.js
+++ b/app/assets/javascripts/tree.js
@@ -1,6 +1,6 @@
 /* eslint-disable func-names, space-before-function-paren, wrap-iife, max-len, quotes, consistent-return, no-var, one-var, one-var-declaration-per-line, no-else-return, prefer-arrow-callback, class-methods-use-this */
 
-class TreeView {
+export default class TreeView {
   constructor() {
     this.initKeyNav();
     // Code browser tree slider
@@ -62,5 +62,3 @@ class TreeView {
     });
   }
 }
-
-window.TreeView = TreeView;