fix undefined method `shift' error on migration