diff --git a/spec/controllers/api/shipments_controller_spec.rb b/spec/controllers/api/shipments_controller_spec.rb
index 9e838617e3..659e1dc761 100644
--- a/spec/controllers/api/shipments_controller_spec.rb
+++ b/spec/controllers/api/shipments_controller_spec.rb
@@ -214,7 +214,7 @@ describe Api::ShipmentsController, type: :controller do
     end
 
     def make_order_contents_fail
-      expect(Spree::Order).to receive(:find_by_number!) { order }
+      expect(Spree::Order).to receive(:find_by!).with({ number: order.number }) { order }
       expect(order).to receive(:contents) { raise error_message }
     end