diff --git a/colabs/boosting/Simple_LightGBM_Integration.ipynb b/colabs/boosting/Simple_LightGBM_Integration.ipynb
index 263e64df..95222f8f 100644
--- a/colabs/boosting/Simple_LightGBM_Integration.ipynb
+++ b/colabs/boosting/Simple_LightGBM_Integration.ipynb
@@ -111,7 +111,7 @@
    "outputs": [],
    "source": [
     "import wandb\n",
-    "from wandb.lightgbm import wandb_callback, log_summary\n"
+    "from wandb.integration.lightgbm import wandb_callback, log_summary\n"
    ]
   },
   {