Fix migration test