Fixing VMID assignment bug