Fix migration error