diff --git a/src/main/java/seedu/address/model/AddressBook.java b/src/main/java/seedu/address/model/AddressBook.java
index 3af09dd889..9dee5ea5c0 100644
--- a/src/main/java/seedu/address/model/AddressBook.java
+++ b/src/main/java/seedu/address/model/AddressBook.java
@@ -303,6 +303,7 @@ public void setProduct(Product target, Product editedProduct) {
                 currOrder.setCreationDate(orderList.getValue().getCreationDate());
                 currOrder.setDeadline(orderList.getValue().getDeadlineObject());
                 currOrder.setCustomer(orderList.getValue().getCustomer());
+                currOrder.setStageContext(orderList.getValue().getStageContext());
             }
             updatedOrders.addOrderWithID(currOrder, currOrder.getId());
         }