Adjust migration for heroku