diff --git a/notebooks/regression_in_keras.ipynb b/notebooks/regression_in_keras.ipynb
index 5f26ad5..1fdc719 100644
--- a/notebooks/regression_in_keras.ipynb
+++ b/notebooks/regression_in_keras.ipynb
@@ -40,7 +40,7 @@
     "from keras.datasets import boston_housing\n",
     "from keras.models import Sequential\n",
     "from keras.layers import Dense, Dropout \n",
-    "from keras.layers.normalization import BatchNormalization "
+    "from keras.layers import BatchNormalization "
    ]
   },
   {