Migration fix for heroku