Fix migrate task