diff --git a/pystencils/sympyextensions.py b/pystencils/sympyextensions.py index f788dbcd89850503e3116598e48b359b037eebb7..8f65e236ba5e59f16c62642208bc1dde349b2b56 100644 --- a/pystencils/sympyextensions.py +++ b/pystencils/sympyextensions.py @@ -260,8 +260,8 @@ def subs_additive(expr: sp.Expr, replacement: sp.Expr, subexpression: sp.Expr, if not param_list: return current_expr else: - if current_expr.func == sp.Mul and sp.numbers.Zero() in param_list: - return sp.numbers.Zero() + if current_expr.func == sp.Mul and sp.core.numbers.Zero() in param_list: + return sp.core.numbers.Zero() else: return current_expr.func(*param_list, evaluate=False)