Fix vm.py imports