diff --git a/site/en/tutorials/audio/music_generation.ipynb b/site/en/tutorials/audio/music_generation.ipynb
index e1423ef7cf..0667c761f7 100644
--- a/site/en/tutorials/audio/music_generation.ipynb
+++ b/site/en/tutorials/audio/music_generation.ipynb
@@ -999,7 +999,7 @@
       "source": [
         "callbacks = [\n",
         "    tf.keras.callbacks.ModelCheckpoint(\n",
-        "        filepath='./training_checkpoints/ckpt_{epoch}',\n",
+        "        filepath='./training_checkpoints/ckpt_{epoch}.weights.h5',\n",
         "        save_weights_only=True),\n",
         "    tf.keras.callbacks.EarlyStopping(\n",
         "        monitor='loss',\n",