Correcting DB migration issue