Fix to migration for heroku