Upgrade this gem to support Rails 3.1

Thanks Austin!