Fix test_migrate_with_volume import.
Merge "Fix test_migrate_with_volume import."