Fixes test migration.