diff --git a/rv32.sv b/rv32.sv
index 8e43ee2..945525f 100644
--- a/rv32.sv
+++ b/rv32.sv
@@ -18,7 +18,7 @@ module rv32 (
 
     rv32_hazard hazard (
         /* control in */
-        .branch_taken_in(mem_branch_taken),
+        .mem_branch_taken_in(mem_branch_taken),
 
         /* control out */
         .fetch_stall_out(fetch_stall),
diff --git a/rv32_hazard.sv b/rv32_hazard.sv
index adc147f..ebcb400 100644
--- a/rv32_hazard.sv
+++ b/rv32_hazard.sv
@@ -3,7 +3,7 @@
 
 module rv32_hazard (
     /* control in */
-    input branch_taken_in,
+    input mem_branch_taken_in,
 
     /* control out */
     output fetch_stall_out,
@@ -22,10 +22,10 @@ module rv32_hazard (
     assign fetch_flush_out = 0;
 
     assign decode_stall_out = execute_stall_out;
-    assign decode_flush_out = fetch_stall_out || branch_taken_in;
+    assign decode_flush_out = fetch_stall_out || mem_branch_taken_in;
 
     assign execute_stall_out = mem_stall_out;
-    assign execute_flush_out = decode_stall_out || branch_taken_in;
+    assign execute_flush_out = decode_stall_out || mem_branch_taken_in;
 
     assign mem_stall_out = 0;
     assign mem_flush_out = execute_stall_out;