diff --git a/demo/Main.py b/demo/Main.py
index 8148d3e6..1445ab59 100644
--- a/demo/Main.py
+++ b/demo/Main.py
@@ -2881,7 +2881,7 @@ class MySplashScreen(SplashScreen):
 
 from wx.lib.mixins.treemixin import ExpansionState
 if USE_CUSTOMTREECTRL:
-    import wx.lib.customtreectrl as CT
+    import wx.lib.agw.customtreectrl as CT
     TreeBaseClass = CT.CustomTreeCtrl
 else:
     TreeBaseClass = wx.TreeCtrl