diff --git a/utils/image_utils.py b/utils/image_utils.py
index 7be6f7d..91435a7 100644
--- a/utils/image_utils.py
+++ b/utils/image_utils.py
@@ -1,6 +1,7 @@
 import sys
 from utils import random_utils
 import numpy as np
+import tensorflow as tf
 import cv2
 import localization
 from scipy.spatial import Delaunay
@@ -294,4 +295,4 @@ class TFLabConverter():
     def lab2bgr(self, lab):    
         return self.tf_sess.run(self.bgr_output_tensor, feed_dict={self.lab_input_tensor: lab})    
         
-    
\ No newline at end of file
+