diff --git a/spec/controllers/checkout_controller_spec.rb b/spec/controllers/checkout_controller_spec.rb
index b373ae64b5..c5d86ace69 100644
--- a/spec/controllers/checkout_controller_spec.rb
+++ b/spec/controllers/checkout_controller_spec.rb
@@ -223,7 +223,8 @@ describe CheckoutController, type: :controller do
     let!(:payment_failed) { create(:payment, order: order, state: 'failed') }
 
     before do
-      order.update_attribute(:shipping_method_id, shipment_pending.shipping_method_id)
+      order.shipments << shipment_pending
+      order.save
       controller.instance_variable_set(:@order, order.reload)
     end