Fix migration command tests.