Update restore_vm.py