Fix issue with migration failure to heroku