diff --git a/for_3b1b_videos/pi_creature.py b/for_3b1b_videos/pi_creature.py
index ea688c46..e0f53707 100644
--- a/for_3b1b_videos/pi_creature.py
+++ b/for_3b1b_videos/pi_creature.py
@@ -1,4 +1,5 @@
 import numpy as np
+import warnings
 
 from constants import *