diff --git a/tests/keras_contrib/layers/test_advanced_activations.py b/tests/keras_contrib/layers/test_advanced_activations.py
index 2254993..0dbfb69 100644
--- a/tests/keras_contrib/layers/test_advanced_activations.py
+++ b/tests/keras_contrib/layers/test_advanced_activations.py
@@ -1,19 +1,17 @@
 import pytest
 from keras.utils.test_utils import layer_test, keras_test
-from keras_contrib import layers
+from keras_contrib.layers import advanced_activations
 
 
 @keras_test
 def test_pelu():
-    from keras_contrib.layers.advanced_activations import PELU
-    layer_test(PELU, kwargs={},
+    layer_test(advanced_activations.PELU, kwargs={},
                input_shape=(2, 3, 4))
 
 
 @keras_test
 def test_pelu_share():
-    from keras_contrib.layers.advanced_activations import PELU
-    layer_test(PELU, kwargs={'shared_axes': 1},
+    layer_test(advanced_activations.PELU, kwargs={'shared_axes': 1},
                input_shape=(2, 3, 4))