fix: Make it work with Rails 3