diff --git a/fawkes/differentiator.py b/fawkes/differentiator.py
index 682028b..474da50 100644
--- a/fawkes/differentiator.py
+++ b/fawkes/differentiator.py
@@ -9,7 +9,7 @@ import time
 import numpy as np
 import tensorflow as tf
 from fawkes.utils import preprocess, reverse_preprocess
-from keras.utils import Progbar
+from tensorflow.keras.utils import Progbar
 
 
 class FawkesMaskGeneration:
diff --git a/fawkes/utils.py b/fawkes/utils.py
index 07ccf36..bb04062 100644
--- a/fawkes/utils.py
+++ b/fawkes/utils.py
@@ -21,7 +21,7 @@ import zipfile
 import PIL
 import pkg_resources
 import six
-from keras.utils import Progbar
+from tensorflow.keras.utils import Progbar
 from six.moves.urllib.error import HTTPError, URLError
 
 stderr = sys.stderr