diff --git a/scripts/analyze_training_multi.py b/scripts/analyze_training_multi.py
index 23f6842..b86218d 100644
--- a/scripts/analyze_training_multi.py
+++ b/scripts/analyze_training_multi.py
@@ -10,6 +10,7 @@
 import numpy as np
 
 import dream.analysis as dream_analysis
+import dream.utilities
 
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~