Updating the MigrationShellTest to reflect the changes to the MigrationShell