chore: fix safe_math

AztecProtocol · Aug 24, 2023 · b4ca535 · b4ca535
1 parent fd6dc4f
commit b4ca535
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/yarn-project/noir-contracts/src/contracts/lending_contract/src/safe_math.nr b/yarn-project/noir-contracts/src/contracts/lending_contract/src/safe_math.nr
@@ -36,10 +36,11 @@ impl SafeU120 {
         b: SafeU120,
     ) -> SafeU120 {
         let c: u120 = self.value * b.value;
-        assert(c / b.value == self.value);
-        SafeU120 {
-            value: c
+        if b.value > 0 {
+            // Only check if b is non-zero to not run into division by zero issues.
+            assert(c / b.value == self.value);
         }
+        SafeU120 { value: c }
     }
 
     fn div(