More migration updates for Heroku