Fix db migration