1 files changed, 3 insertions, 1 deletions
diff --git a/validate.py b/validate.py
index 2e46a9b..9950d07 100755
--- a/validate.py
+++ b/validate.py
@@ -363,7 +363,9 @@ def validate_opi(inputs):
     if funct[2] in ["sll", "srl", "sra"]:
         try:
             shamt = pop_immediate(inputs, 5)
-            mode = pop_immediate(inputs, 7)
+            mode = validate_part(
+                inputs, ["funct7", "mode"], {"norm": 0, "alt": 32}, NAME="comp-mode"
+            )
         except ValueError:
             imm = pop_immediate(inputs, 12)
             shamt = ((imm[0] & 0b11111), 'imm5')