Minor fix on the migrate module