Correct __iadd__() function

6 years ago · 6178befedd
parent d0c00eef9c
commit 6178befedd
1 changed files with 6 additions and 2 deletions
--- a/mythril/laser/ethereum/state.py
+++ b/mythril/laser/ethereum/state.py
@ -167,8 +167,12 @@ class Constraints(list):
            new_solver.add(constraint)
        return Constraints(constraint_list=constraints_list, solver=new_solver)

-    def __iadd__(self, other):
-        return self.__add__(other)
+    def __iadd__(self, constraints):
+        super(Constraints, self).__iadd__(constraints)
+        self.solver = copy(self.solver)
+        for constraint in constraints:
+            self.solver.add(constraint)
+        return self


 class MachineStack(list):