Update vm_helper.py