Fix Heroku migration issue