Another fix for heroku migration