diff --git a/17_autoencoders_gans_and_diffusion_models.ipynb b/17_autoencoders_gans_and_diffusion_models.ipynb
index 28188ac..b3fb4cb 100644
--- a/17_autoencoders_gans_and_diffusion_models.ipynb
+++ b/17_autoencoders_gans_and_diffusion_models.ipynb
@@ -2262,7 +2262,7 @@
     "dataset = tf.data.Dataset.from_tensor_slices(X_train_dcgan)\n",
     "dataset = dataset.shuffle(1000)\n",
     "dataset = dataset.batch(batch_size, drop_remainder=True).prefetch(1)\n",
-    "train_gan(gan, dataset, batch_size, codings_size)"
+    "train_gan(gan, dataset, batch_size, codings_size, n_epochs=50)"
    ]
   },
   {