cleeanup

a16z · Nov 29, 2023 · 6db666b · 6db666b
1 parent 6e5f1fe
commit 6db666b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/halmos/sevm.py b/src/halmos/sevm.py
@@ -1227,11 +1227,11 @@ def decode(cls, loc: Any) -> Any:
             lo = cls.decode(simplify(Extract(255, 0, args)))
             return cls.simple_hash(Concat(hi, lo))
         elif loc.decl().name().startswith("sha3_"):
-            arg = cls.normalize(loc.arg(0))
-            if arg.decl().name() == "concat":
-                return cls.simple_hash(concat([cls.decode(arg.arg(i)) for i in range(arg.num_args())]))
+            sha3_input = cls.normalize(loc.arg(0))
+            if sha3_input.decl().name() == "concat":
+                return cls.simple_hash(concat([cls.decode(sha3_input.arg(i)) for i in range(sha3_input.num_args())]))
             else:
-                return cls.simple_hash(cls.decode(arg))
+                return cls.simple_hash(cls.decode(sha3_input))
         elif loc.decl().name() == "bvadd":
             args = loc.children()
             if len(args) < 2:
@@ -1243,7 +1243,8 @@ def decode(cls, loc: Any) -> Any:
                 return cls.add_all([cls.simple_hash(con(preimage)), con(delta)])
             else:
                 return loc
-        elif is_bv(loc):
+
+        if is_bv(loc):
             return loc
         else:
             raise ValueError(loc)