diff --git a/colabs/intro/Intro_to_Weights_&_Biases.ipynb b/colabs/intro/Intro_to_Weights_&_Biases.ipynb
index 9022a574..8725dbcc 100644
--- a/colabs/intro/Intro_to_Weights_&_Biases.ipynb
+++ b/colabs/intro/Intro_to_Weights_&_Biases.ipynb
@@ -110,6 +110,7 @@
    "outputs": [],
    "source": [
     "import random\n",
+    "import math\n",
     "\n",
     "# Launch 5 simulated experiments\n",
     "total_runs = 5\n",