diff --git a/modules/purchase_amendment/purchase.py b/modules/purchase_amendment/purchase.py index c64321d5a1..75bb9f2f21 100644 --- a/modules/purchase_amendment/purchase.py +++ b/modules/purchase_amendment/purchase.py @@ -161,7 +161,7 @@ class Amendment(Workflow, ModelSQL, ModelView): @classmethod def _stock_moves(cls, line): for move in line.moves: - if move.state == 'draft': + if move.state in {'staging', 'draft'}: yield move diff --git a/modules/sale_amendment/sale.py b/modules/sale_amendment/sale.py index 43353dfc0a..5758527214 100644 --- a/modules/sale_amendment/sale.py +++ b/modules/sale_amendment/sale.py @@ -177,7 +177,7 @@ class Amendment(Workflow, ModelSQL, ModelView): @classmethod def _stock_moves(cls, line): for move in line.moves: - if move.state == 'draft': + if move.state in {'staging', 'draft'}: yield move