1 files changed, 4 insertions, 4 deletions
diff --git a/src/tb/tb_sha512_core.v b/src/tb/tb_sha512_core.v
index e2b8a21..955fee6 100644
--- a/src/tb/tb_sha512_core.v
+++ b/src/tb/tb_sha512_core.v
@@ -422,22 +422,22 @@ module tb_sha512_core();
      case (mode)
        MODE_SHA_512_224:
          begin
-           mask = {{7{32'hffffffff}}, {9{32'h00000000}}};
+           mask = {{224{1'h1}}, 288'h0};
          end
 
        MODE_SHA_512_256:
          begin
-           mask = {{8{32'hffffffff}}, {8{32'h00000000}}};
+           mask = {{256{1'h1}}, 256'h0};
          end
 
        MODE_SHA_384:
          begin
-           mask = {{12{32'hffffffff}}, {4{32'h00000000}}};
+           mask = {{384{1'h1}}, 128'h0};
          end
 
        MODE_SHA_512:
          begin
-           mask = {16{32'hffffffff}};
+           mask = {512{1'h1}};
          end
      endcase // case (mode)