diff --git a/6.1-using-word-embeddings.ipynb b/6.1-using-word-embeddings.ipynb
index 8596b2a218..fbb35957fb 100644
--- a/6.1-using-word-embeddings.ipynb
+++ b/6.1-using-word-embeddings.ipynb
@@ -239,7 +239,7 @@
    ],
    "source": [
     "from keras.models import Sequential\n",
-    "from keras.layers import Flatten, Dense\n",
+    "from keras.layers import Flatten, Dense, Embedding\n",
     "\n",
     "model = Sequential()\n",
     "# We specify the maximum input length to our Embedding layer\n",