fix broken migration file (manually edited it before)
quick refactor with tips from jason