X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/7d36c6a8eb8b1a7d3f62087efa80d9075c5c5e5c..75d9e502238df95fa53c0030b1fac0d4f43e4025:/wxPython/tests/TreeMixinTest.py

diff --git a/wxPython/tests/TreeMixinTest.py b/wxPython/tests/TreeMixinTest.py
index 1153f0243f..ac7909836e 100644
--- a/wxPython/tests/TreeMixinTest.py
+++ b/wxPython/tests/TreeMixinTest.py
@@ -1,4 +1,9 @@
-import wx, wx.gizmos, wx.lib.customtreectrl, unittest, treemixin
+import wx, wx.gizmos, wx.lib.customtreectrl, unittest
+try:
+    import treemixin 
+except ImportError:
+    from wx.lib.mixins import treemixin
+
 
 
 # VirtualTree tests