Fixed db migration issue