diff --git a/layers/BBB/BBBConv.py b/layers/BBB/BBBConv.py
index 1e90103..d8b39c4 100755
--- a/layers/BBB/BBBConv.py
+++ b/layers/BBB/BBBConv.py
@@ -18,7 +18,7 @@ def __init__(self, in_channels, out_channels, kernel_size,
         super(BBBConv2d, self).__init__()
         self.in_channels = in_channels
         self.out_channels = out_channels
-        self.kernel_size = (kernel_size, kernel_size)
+        self.kernel_size = kernel_size if isinstance(kernel_size, tuple) else (kernel_size, kernel_size)
         self.stride = stride
         self.padding = padding
         self.dilation = dilation