diff --git a/spec/models/gateway/braintree_gateway_spec.rb b/spec/models/gateway/braintree_gateway_spec.rb
index e48c74d2..ee1aeeda 100644
--- a/spec/models/gateway/braintree_gateway_spec.rb
+++ b/spec/models/gateway/braintree_gateway_spec.rb
@@ -80,7 +80,7 @@
 
   describe 'payment profile failure' do
     before do
-      country = Spree::Country.default
+      country = Spree::Store.default.default_country
       state   = country.states.first
       address = create(:address,
         firstname: 'John',