Fix migration task test