diff --git a/wxPython/src/_notebook.i b/wxPython/src/_notebook.i
index 39bf218ea0..e654aa9ffa 100644
--- a/wxPython/src/_notebook.i
+++ b/wxPython/src/_notebook.i
@@ -529,7 +529,7 @@ public:
     int GetPageParent(size_t pos) const;
 
     // the tree control we use for showing the pages index tree
-    wxTreeCtrl* GetTreeCtrl() const;
+    wxPyTreeCtrl* GetTreeCtrl() const;
 
 };
 
diff --git a/wxPython/src/html.i b/wxPython/src/html.i
index f272833ff8..3212eb0620 100644
--- a/wxPython/src/html.i
+++ b/wxPython/src/html.i
@@ -1373,6 +1373,9 @@ enum {
 };
 
 
+%{
+    typedef wxTreeCtrl wxPyTreeCtrl;
+%}
 
 MustHaveApp(wxHtmlHelpWindow);
 
@@ -1464,7 +1467,7 @@ public:
     wxHtmlHelpFrameCfg& GetCfgData();
 
     // Gets the tree control
-    wxTreeCtrl *GetTreeCtrl() const;
+    wxPyTreeCtrl *GetTreeCtrl() const;
 
 };