diff --git a/spec/models/spree/order_spec.rb b/spec/models/spree/order_spec.rb
index 1b12fcbc3d..7fa0f365c1 100644
--- a/spec/models/spree/order_spec.rb
+++ b/spec/models/spree/order_spec.rb
@@ -229,10 +229,12 @@ RSpec.describe Spree::Order do
     end
 
     it "should sell inventory units" do
-      order.shipments.each do |shipment|
-        expect(shipment).to receive(:update!)
-        expect(shipment).to receive(:finalize!)
-      end
+      shipment = Spree::Shipment.new
+      order.shipments = [shipment]
+
+      expect(shipment).to receive(:update!).with(order)
+      expect(shipment).to receive(:finalize!)
+
       order.finalize!
     end