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: