Fix migration file