```ruby
module Spree
  module Admin
    class StatesController < ResourceController
      belongs_to :country
      before_filter :load_data

      def index
        respond_with(@collection) do |format|
          format.html
          format.js { render :partial => 'state_list' }
        end
      end

      protected

      def location_after_save
        admin_country_states_url(@country)
      end

      def collection
        super.order(:name)
      end

      def load_data
        @countries = Country.order(:name)
      end
    end
  end
end
```

Correct reference to Spree::Country in Admin::StatesController
user
```ruby
module Spree
  module Admin
    class StatesController < ResourceController
      belongs_to 'spree/country'
      before_filter :load_data

      def index
        respond_with(@collection) do |format|
          format.html
          format.js { render :partial => 'state_list' }
        end
      end

      protected

      def location_after_save
        admin_country_states_url(@country)
      end

      def collection
        super.order(:name)
      end

      def load_data
        @countries = Country.order(:name)
      end
    end
  end
end
```
assistant