diff --git a/ppcls/arch/gears/bnneck.py b/ppcls/arch/gears/bnneck.py
index 8e97409efa2bb6629b7c5f742bb202e917534d9b..15b6f46b05a7c0390257b1882d006612cef4390d 100644
--- a/ppcls/arch/gears/bnneck.py
+++ b/ppcls/arch/gears/bnneck.py
@@ -1,4 +1,7 @@
-class BNNeck(nn.Layer):
+import paddle
+
+
+class BNNeck(paddle.nn.Layer):
     def __init__(self, num_filters, trainable=False):
         super(BNNeck, self).__init__()
         self.num_filters = num_filters