diff --git a/lib/generators/my_zipcode_gem/templates/zipcode_model.rb b/lib/generators/my_zipcode_gem/templates/zipcode_model.rb
index <HASH>..<HASH> 100644
--- a/lib/generators/my_zipcode_gem/templates/zipcode_model.rb
+++ b/lib/generators/my_zipcode_gem/templates/zipcode_model.rb
@@ -13,6 +13,7 @@ class Zipcode < ActiveRecord::Base
     def find_by_city_state(city, state)
       includes(county: :state)
         .where("city like ? AND states.abbr like ?", "#{city}%", "%#{state}%")
+        .references(:state)
         .first
     end
   end
fix(zipcode): Fix zipcode in Rails 4+

Close #<I>