Update for new Spree version