Rails 4 fix