Fix heroku migration