Small fix for Rails 4