diff --git a/courses/udacity_intro_to_tensorflow_for_deep_learning/l05c01_dogs_vs_cats_without_augmentation.ipynb b/courses/udacity_intro_to_tensorflow_for_deep_learning/l05c01_dogs_vs_cats_without_augmentation.ipynb
index 24503553d88..586c1e2f419 100644
--- a/courses/udacity_intro_to_tensorflow_for_deep_learning/l05c01_dogs_vs_cats_without_augmentation.ipynb
+++ b/courses/udacity_intro_to_tensorflow_for_deep_learning/l05c01_dogs_vs_cats_without_augmentation.ipynb
@@ -230,7 +230,7 @@
       },
       "outputs": [],
       "source": [
-        "base_dir = os.path.join(os.path.dirname(zip_dir), 'cats_and_dogs_filtered')\n",
+        "base_dir = os.path.join(zip_dir, 'cats_and_dogs_filtered')\n",
         "train_dir = os.path.join(base_dir, 'train')\n",
         "validation_dir = os.path.join(base_dir, 'validation')\n",
         "\n",
@@ -591,7 +591,7 @@
       "outputs": [],
       "source": [
         "EPOCHS = 100\n",
-        "history = model.fit_generator(\n",
+        "history = model.fit(\n",
         "    train_data_gen,\n",
         "    steps_per_epoch=int(np.ceil(total_train / float(BATCH_SIZE))),\n",
         "    epochs=EPOCHS,\n",