diff --git a/keras_visualize_activations/read_activations.py b/keras_visualize_activations/read_activations.py
index b5d34ab9cfe77e1a6240d3175e2c37a7fc52b2a8..0e4641ac32bfcd48aa42cefb676d021e27697310 100644
--- a/keras_visualize_activations/read_activations.py
+++ b/keras_visualize_activations/read_activations.py
@@ -1,6 +1,5 @@
 import keras.backend as K
 
-from keras.layers import Masking, InputLayer
 
 def get_activations(model, model_inputs, print_shape_only=False, layer_name=None):
     print('----- activations -----')
@@ -13,12 +12,8 @@ def get_activations(model, model_inputs, print_shape_only=False, layer_name=None
         inp = [inp]
         model_multi_inputs_cond = False
 
-    # all layer outputs
-    # skip input and masking layers
     outputs = [layer.output for layer in model.layers if
-               (layer.name == layer_name or layer_name is None)
-               and not isinstance(layer, InputLayer)
-               and not isinstance(layer, Masking)]
+               layer.name == layer_name or layer_name is None]  # all layer outputs
 
     funcs = [K.function(inp + [K.learning_phase()], [out]) for out in outputs]  # evaluation functions