diff --git a/10_neural_nets_with_keras.ipynb b/10_neural_nets_with_keras.ipynb
index caeac36..8c1ca66 100644
--- a/10_neural_nets_with_keras.ipynb
+++ b/10_neural_nets_with_keras.ipynb
@@ -741,7 +741,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "y_pred = model.predict_classes(X_new)\n",
+    "y_pred = np.argmax(model.predict(X_new), axis=-1)\n",
     "y_pred"
    ]
   },