Fix migration problem