Fix migration process