diff --git a/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb b/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb
index 6c3ada81..7280dbec 100644
--- a/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb
+++ b/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb
@@ -160,7 +160,7 @@
     "import random\n",
     "import numpy as np\n",
     "import tensorflow as tf\n",
-    "from wandb.keras import WandbMetricsLogger, WandbModelCheckpoint\n",
+    "from wandb.integration.keras import WandbMetricsLogger, WandbModelCheckpoint\n",
     "\n",
     "# Simple Keras Model\n",
     "\n",
@@ -209,7 +209,7 @@
     "    # Add WandbMetricsLogger to log metrics and WandbModelCheckpoint to log model checkpoints\n",
     "    wandb_callbacks = [\n",
     "        WandbMetricsLogger(),\n",
-    "        WandbModelCheckpoint(filepath=\"my_model_{epoch:02d}\"),\n",
+    "        WandbModelCheckpoint(filepath=\"my_model_{epoch:02d}.keras\"),\n",
     "    ]\n",
     "\n",
     "    model.fit(\n",