Changed the migration according to the comments.