diff --git a/pystencils/sympyextensions.py b/pystencils/sympyextensions.py index e40c108959ed346c37e65092fd767f175b2e16d5..3dc84f146474b9b708637205d4a625ba921b8cc8 100644 --- a/pystencils/sympyextensions.py +++ b/pystencils/sympyextensions.py @@ -250,7 +250,10 @@ def subs_additive(expr: sp.Expr, replacement: sp.Expr, subexpression: sp.Expr, if not param_list: return current_expr else: - return current_expr.func(*param_list, evaluate=False) + if current_expr.func == sp.Mul and sp.numbers.Zero() in param_list: + return sp.numbers.Zero() + else: + return current_expr.func(*param_list, evaluate=False) return visit(expr)