diff --git a/Orange/widgets/visualize/owtreeviewer.py b/Orange/widgets/visualize/owtreeviewer.py
index ceb88fbe0d1..ae48d45c85f 100644
--- a/Orange/widgets/visualize/owtreeviewer.py
+++ b/Orange/widgets/visualize/owtreeviewer.py
@@ -263,6 +263,7 @@ def ctree(self, model=None):
         self.color_combo.clear()
         self.closeContext()
         self.model = model
+        self.target_class_index = 0
         if model is None:
             self.info.setText('No tree.')
             self.root_node = None
@@ -440,7 +441,7 @@ def _get_tree_adapter(self, model):
         return TreeAdapter(model)
 
 
-def test():
+def main():
     """Standalone test"""
     import sys
     from AnyQt.QtWidgets import QApplication
@@ -458,4 +459,4 @@ def test():
     ow.saveSettings()
 
 if __name__ == "__main__":
-    test()
+    main()