modifying migration for heroku
fix migration code