diff --git a/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py b/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py
index 251d85c..ac5a82e 100644
--- a/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py
+++ b/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py
@@ -13,7 +13,7 @@
 import pandas as pd
 from sklearn.linear_model import LogisticRegressionCV
 from sklearn.ensemble import RandomForestClassifier
-from sklearn.cross_validation import StratifiedShuffleSplit
+from sklearn.model_selection import StratifiedShuffleSplit
 from sklearn.preprocessing import StandardScaler
 from sklearn.metrics import roc_auc_score
 from sklearn.tree import DecisionTreeClassifier