diff --git a/examples/example_with_scenes.ipynb b/examples/example_with_scenes.ipynb
index 595c52c9..83d6afc2 100644
--- a/examples/example_with_scenes.ipynb
+++ b/examples/example_with_scenes.ipynb
@@ -71,6 +71,7 @@
         "# @title Imports\n",
         "\n",
         "import collections\n",
+        "from collections.abc import Callable, Mapping, Sequence\n",
         "import concurrent.futures\n",
         "import datetime\n",
         "\n",
@@ -91,6 +92,8 @@
         "from concordia.components import game_master as gm_components\n",
         "from concordia.environment import game_master\n",
         "from concordia.environment.scenes import runner\n",
+        "from concordia.language_model import gpt_model\n",
+        "from concordia.language_model import gcloud_model\n",
         "from concordia.language_model import language_model\n",
         "from concordia.metrics import goal_achievement\n",
         "from concordia.metrics import common_sense_morality\n",