Small change on migrate