diff --git a/07_ensemble_learning_and_random_forests.ipynb b/07_ensemble_learning_and_random_forests.ipynb
index f90425019..7ec083e96 100644
--- a/07_ensemble_learning_and_random_forests.ipynb
+++ b/07_ensemble_learning_and_random_forests.ipynb
@@ -146,7 +146,7 @@
     }
    ],
    "source": [
-    "from sklearn.cross_validation import train_test_split\n",
+    "from sklearn.model_selection import train_test_split\n",
     "from sklearn.datasets import make_moons\n",
     "\n",
     "X, y = make_moons(n_samples=500, noise=0.30, random_state=42)\n",
@@ -906,7 +906,7 @@
    },
    "outputs": [],
    "source": [
-    "from sklearn.cross_validation import train_test_split\n",
+    "from sklearn.model_selection import train_test_split\n",
     "from sklearn.metrics import mean_squared_error\n",
     "\n",
     "X_train, X_val, y_train, y_val = train_test_split(X, y)\n",
@@ -1080,7 +1080,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.1"
+   "version": "3.5.2"
   },
   "nav_menu": {
    "height": "252px",