Issue when adding a counter cache column

The rate_average function was returning an infinity when using rates_count cached column. It was using the old number before the model updated it. I fixed it by adding the following code to the update_cached_average function:

rates(:refresh).size if self.respond_to?(:rates_count)

full function is now:

def update_cached_average
  if self.class.caching_average?
    rates(:refresh).size if self.respond_to?(:rates_count)
    send("#{self.class.options[:cache_column]}=", self.rate_average(false))
    save!
  end
end

end