Fix db migrate task