diff --git a/libs/pconv_layer.py b/libs/pconv_layer.py
index b6fdfe9..1afb6a5 100644
--- a/libs/pconv_layer.py
+++ b/libs/pconv_layer.py
@@ -33,7 +33,7 @@ def build(self, input_shape):
                                       regularizer=self.kernel_regularizer,
                                       constraint=self.kernel_constraint)
         # Mask kernel
-        self.kernel_mask = K.ones(shape=self.kernel_size + (self.input_dim, self.filters))
+        self.kernel_mask = K.constant(1, shape=self.kernel_size + (self.input_dim, self.filters))
 
         # Calculate padding size to achieve zero-padding
         self.pconv_padding = (