1 files changed, 3 insertions, 0 deletions
diff --git a/src/tb/tb_sha256_core.v b/src/tb/tb_sha256_core.v
index 07e2334..981351c 100644
--- a/src/tb/tb_sha256_core.v
+++ b/src/tb/tb_sha256_core.v
@@ -68,6 +68,7 @@ module tb_sha256_core();
   reg            tb_reset_n;
   reg            tb_init;
   reg            tb_next;
+  reg            tb_mode;
   reg [511 : 0]  tb_block;
   reg  [31 : 0]  tb_state_wr_data;
   reg            tb_state0_we;
@@ -92,6 +93,7 @@ module tb_sha256_core();
 
                   .init(tb_init),
                   .next(tb_next),
+                  .mode(tb_mode),
 
                   .block(tb_block),
 
@@ -244,6 +246,7 @@ module tb_sha256_core();
 
       tb_init          = 0;
       tb_next          = 0;
+      tb_mode          = 1;
       tb_block         = 512'h0;
       tb_state_wr_data = 32'h0;
       tb_state0_we     = 0;