diff --git a/scripts/ai-heatmap.ipynb b/scripts/ai-heatmap.ipynb
index 589a6bb..3b2a707 100644
--- a/scripts/ai-heatmap.ipynb
+++ b/scripts/ai-heatmap.ipynb
@@ -31,6 +31,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "!pip install matplotlib numpy pytz\n",
+    "\n",
     "import json\n",
     "\n",
     "with open(f'{convo_folder}/conversations.json', 'r') as f:\n",