diff --git a/wario/contact_features.ipynb b/wario/contact_features.ipynb
index f63ac11a8a3f6677bc7101bd59e1bfbd93a25d95..5c88a59756e0b870339c908991d470878a7b416e 100644
--- a/wario/contact_features.ipynb
+++ b/wario/contact_features.ipynb
@@ -10,6 +10,7 @@
     "# Load notebooks with required functions\n",
     "import ipynb\n",
     "from ipynb.fs.full.wcontact_matrix import *\n",
+    "from ipynb.fs.full.utils import *\n",
     "\n",
     "# Load required libraries\n",
     "import numpy as np\n",