Fix for migrate issue with Heroku