diff --git a/modules/purchase_amendment/purchase.py b/modules/purchase_amendment/purchase.py index 08d1c67917..af105586d2 100644 --- a/modules/purchase_amendment/purchase.py +++ b/modules/purchase_amendment/purchase.py @@ -153,7 +153,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 0712c66385..93baa1113e 100644 --- a/modules/sale_amendment/sale.py +++ b/modules/sale_amendment/sale.py @@ -169,7 +169,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