diff --git a/code/generate_limit_dataset.ipynb b/code/generate_limit_dataset.ipynb
index 6db0f81..b078f12 100644
--- a/code/generate_limit_dataset.ipynb
+++ b/code/generate_limit_dataset.ipynb
@@ -7,6 +7,7 @@
       "cell_type": "markdown",
       "source": [
         "# LIMIT Dataset Generation\n",
+        "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github//google-deepmind/limit/blob/main/code/generate_limit_dataset.ipynb)\n",
         "\n",
         "This script provides code to generate a LIMIT-style dataset."
       ]