diff --git a/examples/multi_trans.py b/examples/multi_trans.py
index 7a7efc7..4dd7628 100644
--- a/examples/multi_trans.py
+++ b/examples/multi_trans.py
@@ -3,6 +3,7 @@ import warnings
 from diffusers import AutoPipelineForText2Image
 from latentblending.movie_util import concatenate_movies
 from latentblending.blending_engine import BlendingEngine
+import numpy as np
 torch.set_grad_enabled(False)
 torch.backends.cudnn.benchmark = False
 warnings.filterwarnings('ignore')