diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index 66b5017..fbf8e1b 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.ipynb
@@ -543,7 +543,7 @@
    "outputs": [],
    "source": [
     "from sklearn.metrics import mean_squared_error\n",
-    "from sklearn.cross_validation import train_test_split\n",
+    "from sklearn.model_selection import train_test_split\n",
     "\n",
     "def plot_learning_curves(model, X, y):\n",
     "    X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=10)\n",
@@ -1197,7 +1197,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.1"
+   "version": "3.5.2"
   },
   "nav_menu": {},
   "toc": {