From 8405a7672dab5b1dcf7de34e8b7072ab4e85cb67 Mon Sep 17 00:00:00 2001
From: Matt-Yorkley <9029026+Matt-Yorkley@users.noreply.github.com>
Date: Mon, 10 May 2021 22:23:43 +0100
Subject: [PATCH] Bring in shipment updating

---
 app/models/spree/order_contents.rb | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/app/models/spree/order_contents.rb b/app/models/spree/order_contents.rb
index d0478256e6..024038ffec 100644
--- a/app/models/spree/order_contents.rb
+++ b/app/models/spree/order_contents.rb
@@ -13,6 +13,7 @@ module Spree
     def add(variant, quantity = 1, shipment = nil)
       line_item = order.find_line_item_by_variant(variant)
       add_to_line_item(line_item, variant, quantity, shipment)
+      shipment.present? ? shipment.update_amounts : order.ensure_updated_shipments
       update_order
     end
 
@@ -26,6 +27,7 @@ module Spree
       end
 
       remove_from_line_item(line_item, variant, quantity, shipment)
+      shipment.present? ? shipment.update_amounts : order.ensure_updated_shipments
       update_order
     end