Update syntax to be compatible with Rails 4