Fixing db migration