diff --git a/ml/pc/exercises/image_classification_part1.ipynb b/ml/pc/exercises/image_classification_part1.ipynb
index 369acd4..481d947 100644
--- a/ml/pc/exercises/image_classification_part1.ipynb
+++ b/ml/pc/exercises/image_classification_part1.ipynb
@@ -511,7 +511,7 @@
       },
       "outputs": [],
       "source": [
-        "history = model.fit_generator(\n",
+        "history = model.fit(\n",
         "      train_generator,\n",
         "      steps_per_epoch=100,  # 2000 images = batch_size * steps\n",
         "      epochs=15,\n",