diff --git a/BE2_GAN_and_Diffusion.ipynb b/BE2_GAN_and_Diffusion.ipynb
index d0181703393a3d78ef4ef5e81c66b6175e2ba32d..f72c2bd634da870dc9e82d3b3b6070560ec13784 100644
--- a/BE2_GAN_and_Diffusion.ipynb
+++ b/BE2_GAN_and_Diffusion.ipynb
@@ -973,7 +973,7 @@
     "        \n",
     "    if (epoch + 1) % 100 == 0:\n",
     "        print('Saving model...')\n",
-    "        save_model(epoch, loss_D, loss_G)\n"
+    "        save_model(epoch+1, loss_D, loss_G)\n"
    ]
   },
   {