Fix migration issue