#include "kvm_util.h"
 #include "processor.h"
 
-#define VCPU_ID                        1
-
 static void guest_ins_port80(uint8_t *buffer, unsigned int count)
 {
        unsigned long end;
 
 int main(int argc, char *argv[])
 {
+       struct kvm_vcpu *vcpu;
        struct kvm_regs regs;
        struct kvm_run *run;
        struct kvm_vm *vm;
        struct ucall uc;
-       int rc;
 
        /* Tell stdout not to buffer its content */
        setbuf(stdout, NULL);
 
-       /* Create VM */
-       vm = vm_create_default(VCPU_ID, 0, guest_code);
-       run = vcpu_state(vm, VCPU_ID);
+       vm = vm_create_with_one_vcpu(&vcpu, guest_code);
+       run = vcpu->run;
 
        memset(®s, 0, sizeof(regs));
 
        while (1) {
-               rc = _vcpu_run(vm, VCPU_ID);
+               vcpu_run(vm, vcpu->id);
 
-               TEST_ASSERT(rc == 0, "vcpu_run failed: %d\n", rc);
                TEST_ASSERT(run->exit_reason == KVM_EXIT_IO,
                            "Unexpected exit reason: %u (%s),\n",
                            run->exit_reason,
                            exit_reason_str(run->exit_reason));
 
-               if (get_ucall(vm, VCPU_ID, &uc))
+               if (get_ucall(vm, vcpu->id, &uc))
                        break;
 
                TEST_ASSERT(run->io.port == 0x80,
                 * scope from a testing perspective as it's not ABI in any way,
                 * i.e. it really is abusing internal KVM knowledge.
                 */
-               vcpu_regs_get(vm, VCPU_ID, ®s);
+               vcpu_regs_get(vm, vcpu->id, ®s);
                if (regs.rcx == 2)
                        regs.rcx = 1;
                if (regs.rcx == 3)
                        regs.rcx = 8192;
                memset((void *)run + run->io.data_offset, 0xaa, 4096);
-               vcpu_regs_set(vm, VCPU_ID, ®s);
+               vcpu_regs_set(vm, vcpu->id, ®s);
        }
 
        switch (uc.cmd) {