diff --git a/extra/replay_pkl.py b/extra/replay_pkl.py
index 3dde042903..1a6af06de5 100644
--- a/extra/replay_pkl.py
+++ b/extra/replay_pkl.py
@@ -44,13 +44,13 @@ if __name__ == "__main__":
             #elif k.full_shap[-4] == 7: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 7))
             #elif k.full_shape[-4] == 14: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 2))
           elif len(k.full_shape) == 3 and k.full_shape[1] == 32:
-            #if k.full_shape[0]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 0, 4))
+            if k.full_shape[0]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 0, 4))
             # weight without more
             k.apply_opt(Opt(OptOps.UNROLL, 0, 8))
             k.apply_opt(Opt(OptOps.UPCAST, 1, 32))
             if k.full_shape[0]%4 == 0: k.apply_opt(Opt(OptOps.UPCAST, 0, 4))
           elif len(k.full_shape) == 4 and k.full_shape[2] == 32:
-            #if k.full_shape[1]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 1, 4))
+            if k.full_shape[1]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 1, 4))
             # weight with more
             k.apply_opt(Opt(OptOps.UNROLL, 0, 8))
             k.apply_opt(Opt(OptOps.UPCAST, 2, 32))