diff --git a/align_images.py b/align_images.py
index d3ac7ea88..07fc776a4 100644
--- a/align_images.py
+++ b/align_images.py
@@ -1,7 +1,7 @@
 import os
 import sys
 import bz2
-from keras.utils import get_file
+from tensorflow.keras.utils import get_file
 from ffhq_dataset.face_alignment import image_align
 from ffhq_dataset.landmarks_detector import LandmarksDetector