diff --git a/cq_editor/main_window.py b/cq_editor/main_window.py
index d578ca8..5358e96 100644
--- a/cq_editor/main_window.py
+++ b/cq_editor/main_window.py
@@ -7,7 +7,7 @@ import cadquery as cq
 from .widgets.editor import Editor
 from .widgets.viewer import OCCViewer
 from .widgets.console import ConsoleWidget
-from .widgets.object_tree import ObjectTree
+from .widgets.object_tree import ObjectTree, rand_color
 from .widgets.traceback_viewer import TracebackPane
 from .widgets.debugger import Debugger, LocalsView
 from .widgets.cq_object_inspector import CQObjectInspector