diff --git a/Chapter_12_Ensemble_Methods/utils.py b/Chapter_12_Ensemble_Methods/utils.py
index e4131ba..abb0606 100644
--- a/Chapter_12_Ensemble_Methods/utils.py
+++ b/Chapter_12_Ensemble_Methods/utils.py
@@ -47,7 +47,7 @@ def plot_model(X, y, model, fix_margins=True):
     pyplot.show()
 
 def display_tree(dt):
-    from sklearn.externals.six import StringIO  
+    from six import StringIO  
     from IPython.display import Image  
     from sklearn.tree import export_graphviz
     import pydotplus
diff --git a/Chapter_9_Decision_Trees/utils.py b/Chapter_9_Decision_Trees/utils.py
index 4f7c934..719908a 100644
--- a/Chapter_9_Decision_Trees/utils.py
+++ b/Chapter_9_Decision_Trees/utils.py
@@ -39,7 +39,7 @@ def plot_model(X, y, model, size_of_points=100):
     pyplot.show()
     
 def display_tree(dt):
-    from sklearn.externals.six import StringIO  
+    from six import StringIO  
     from IPython.display import Image  
     from sklearn.tree import export_graphviz
     import pydotplus