diff --git a/smallConf.py b/smallConf.py
index 3d75446..a166cec 100644
--- a/smallConf.py
+++ b/smallConf.py
@@ -107,6 +107,7 @@ def nextShape():
         # Network Degree has to be an even number
         if netDegree % 2 == 0:
             blockSizeR = blockSizeC = blockSize
+            blockSizeRK = blockSizeCK = blockSize // 2
             chiR = chiC = chi
-            shape = Shape(blockSizeR, blockSizeC, nn, fm, fr, class1ratio, chiR, chiC, vpn1, vpn2, netDegree, bwUplinkProd, bwUplink1, bwUplink2, run)
+            shape = Shape(blockSizeR, blockSizeRK, blockSizeC, blockSizeCK, nn, fm, fr, class1ratio, chiR, chiC, vpn1, vpn2, netDegree, bwUplinkProd, bwUplink1, bwUplink2, run)
             yield shape