diff --git a/tables/automl/notebooks/energy_price_forecasting/energy_price_forecasting.ipynb b/tables/automl/notebooks/energy_price_forecasting/energy_price_forecasting.ipynb
index fab8984c711..e1f429bb3fc 100644
--- a/tables/automl/notebooks/energy_price_forecasting/energy_price_forecasting.ipynb
+++ b/tables/automl/notebooks/energy_price_forecasting/energy_price_forecasting.ipynb
@@ -202,6 +202,7 @@
       "source": [
         "#@title Authenticate using service account key and create a client. { vertical-output: true }\n",
         "\n",
+        "from google.colab import files\n",
         "from google.cloud import automl_v1beta1\n",
         "\n",
         "# Upload service account key\n",