Update vm_migration.py