lib/restrict_cache/railtie.rb in restrict_cache-0.1.0 vs lib/restrict_cache/railtie.rb in restrict_cache-0.1.1

@@ -7,10 +7,11 @@
       ActiveSupport.on_load(:action_controller) do
         ::ActionController::Base.send(:include, RestrictCache::ActionControllerExt)
       end
 
       ActiveSupport.on_load(:active_record) do
-         ::ActiveRecord::Base.send(:include, RestrictCache::ActiveRecordExt)
+         ::ActiveRecord::Base.send(:include, RestrictCache::ActiveRecordExt::Base)
+         ::ActiveRecord::Relation.send(:include, RestrictCache::ActiveRecordExt::Relation)
       end
     end
   end
 end