Tweak rake script for Heroku