diff --git a/.github/workflows/colab.yml b/.github/workflows/colab.yml
index 7946adfd5d..4ece61ca6a 100644
--- a/.github/workflows/colab.yml
+++ b/.github/workflows/colab.yml
@@ -16,7 +16,7 @@ on:
 
 jobs:
   colab:
-    runs-on: ubuntu-22.04
+    runs-on: ubuntu-latest
     env:
         COLAB_ALWAYS_INSTALL_XVFB: 1
         QLEARNING_NUM_TRAINING_STEPS: 5
diff --git a/colab/Colab_UnityEnvironment_1_Run.ipynb b/colab/Colab_UnityEnvironment_1_Run.ipynb
index e70dca7b8a..9ac80f920f 100644
--- a/colab/Colab_UnityEnvironment_1_Run.ipynb
+++ b/colab/Colab_UnityEnvironment_1_Run.ipynb
@@ -217,7 +217,7 @@
     "\n",
     "from mlagents_envs.registry import default_registry\n",
     "\n",
-    "env = default_registry[env_id].make()"
+    "env = default_registry[env_id].make(no_graphics=True)"
    ],
    "execution_count": 3,
    "outputs": [