diff --git a/keras_visualize_activations/read_activations.py b/keras_visualize_activations/read_activations.py
index 053439b84ebb50be638acb9d158694e2f3c1d9aa..b5d34ab9cfe77e1a6240d3175e2c37a7fc52b2a8 100644
--- a/keras_visualize_activations/read_activations.py
+++ b/keras_visualize_activations/read_activations.py
@@ -1,7 +1,6 @@
 import keras.backend as K
 
-from keras.engine.input_layer import InputLayer
-from keras.layers.core import Masking
+from keras.layers import Masking, InputLayer
 
 def get_activations(model, model_inputs, print_shape_only=False, layer_name=None):
     print('----- activations -----')