diff --git a/colabs/tensorflow/Simple_TensorFlow_Integration.ipynb b/colabs/tensorflow/Simple_TensorFlow_Integration.ipynb
index a42a5a0c..b5f25a49 100644
--- a/colabs/tensorflow/Simple_TensorFlow_Integration.ipynb
+++ b/colabs/tensorflow/Simple_TensorFlow_Integration.ipynb
@@ -136,7 +136,7 @@
    "outputs": [],
    "source": [
     "import wandb\n",
-    "from wandb.keras import WandbCallback\n"
+    "from wandb.integration.keras import WandbCallback"
    ]
   },
   {