Modify migration for heroku