diff --git a/src/ieu/bmu/cnt.sv b/src/ieu/bmu/cnt.sv
index 13ff1e15f..75ace3ac7 100644
--- a/src/ieu/bmu/cnt.sv
+++ b/src/ieu/bmu/cnt.sv
@@ -32,7 +32,7 @@
 
 module cnt #(parameter WIDTH = 32) (
   input  logic [WIDTH-1:0] A, RevA,    // Operands
-  input  logic [4:0] B,                // Last 5 bits of immediate
+  input  logic [1:0] B,                // Last 2 bits of immediate
   input  logic W64,                    // Indicates word operation
   output logic [WIDTH-1:0] CntResult   // count result
 );
diff --git a/src/ieu/bmu/zbb.sv b/src/ieu/bmu/zbb.sv
index 1dff2fd0a..fef8579b5 100644
--- a/src/ieu/bmu/zbb.sv
+++ b/src/ieu/bmu/zbb.sv
@@ -42,7 +42,7 @@ module zbb #(parameter WIDTH=32) (
   logic [WIDTH-1:0] ByteResult;          // byte results
   logic [WIDTH-1:0] ExtResult;           // sign/zero extend results
 
-  cnt #(WIDTH) cnt(.A, .RevA, .B(B[4:0]), .W64, .CntResult);
+  cnt #(WIDTH) cnt(.A, .RevA, .B(B[1:0]), .W64, .CntResult);
   byteUnit #(WIDTH) bu(.A, .ByteSelect(B[0]), .ByteResult);
   ext #(WIDTH) ext(.A, .ExtSelect({~B[2], {B[2] & B[0]}}), .ExtResult);