Add Compatibility for Rails 4.1