Clean up code in GraphMigrator a bit.