diff --git a/efficientnet/model.py b/efficientnet/model.py
index 42371ff..8805a43 100644
--- a/efficientnet/model.py
+++ b/efficientnet/model.py
@@ -33,7 +33,7 @@
 
 from six.moves import xrange
 from keras_applications.imagenet_utils import _obtain_input_shape
-from keras_applications.imagenet_utils import preprocess_input as _preprocess_input
+from tensorflow.keras.applications.imagenet_utils import preprocess_input as _preprocess_input
 
 from . import get_submodules_from_kwargs
 from .weights import IMAGENET_WEIGHTS_PATH, IMAGENET_WEIGHTS_HASHES, NS_WEIGHTS_HASHES, NS_WEIGHTS_PATH