diff --git a/CNN_Fashion_MNIST_Example.ipynb b/CNN_Fashion_MNIST_Example.ipynb
index 831c92c..0130c6f 100644
--- a/CNN_Fashion_MNIST_Example.ipynb
+++ b/CNN_Fashion_MNIST_Example.ipynb
@@ -541,7 +541,7 @@
       },
       "source": [
         "# Get Model Predictions for test data\n",
-        "predicted_classes = model.predict_classes(x_test)\n",
+        "predicted_classes = np.argmax(model.predict(x_test), axis=1)\n",
         "print(classification_report(y_test, predicted_classes, target_names=class_names))"
       ],
       "execution_count": 8,
@@ -691,4 +691,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}