diff --git a/mrcnn/model.py b/mrcnn/model.py
index 62cb2b0951..9096cb589b 100644
--- a/mrcnn/model.py
+++ b/mrcnn/model.py
@@ -17,7 +17,12 @@
 import multiprocessing
 import numpy as np
 import tensorflow as tf
-import keras
+
+#import keras
+from tensorflow.keras import keras.engine as KE
+from tensorflow.keras import layers as KL
+from tensorflow.keras import models as KM
+from tensorflow.keras import backend as K
 import keras.backend as K
 import keras.layers as KL
 import keras.engine as KE