diff --git a/compute_warped_mesh.py b/compute_warped_mesh.py
index cbe4af546cf9929e8d859a2c6f2e833daeafbcd6..ede4424ddc84dbd16e6fbee0c7fd99fce65681ab 100644
--- a/compute_warped_mesh.py
+++ b/compute_warped_mesh.py
@@ -18,6 +18,7 @@ import myPythonLibrary as mypy
 import myVTKPythonLibrary as myvtk
 
 import dolfin_dic as ddic
+from generate_images import *
 
 ################################################################################