diff --git a/colabs/tensorflow/Hyperparameter_Optimization_in_TensorFlow_using_W&B_Sweeps.ipynb b/colabs/tensorflow/Hyperparameter_Optimization_in_TensorFlow_using_W&B_Sweeps.ipynb
index d45de2a0..8da6c2d5 100644
--- a/colabs/tensorflow/Hyperparameter_Optimization_in_TensorFlow_using_W&B_Sweeps.ipynb
+++ b/colabs/tensorflow/Hyperparameter_Optimization_in_TensorFlow_using_W&B_Sweeps.ipynb
@@ -141,7 +141,7 @@
    "outputs": [],
    "source": [
     "import wandb\n",
-    "from wandb.keras import WandbCallback\n"
+    "from wandb.integration.keras import WandbCallback"
    ]
   },
   {