Update for Rails 3.2