Fix migration for testing env.