Extend to Rails 4.0