#include "kvm_util.h"
 #include "processor.h"
 
-#define VCPU_ID                5
-
 #define HCALL_REGION_GPA       0xc0000000ULL
 #define HCALL_REGION_SLOT      10
 
-static struct kvm_vm *vm;
-
 #define INPUTVALUE 17
 #define ARGVALUE(x) (0xdeadbeef5a5a0000UL + x)
 #define RETVALUE 0xcafef00dfbfbffffUL
 
 int main(int argc, char *argv[])
 {
+       struct kvm_vcpu *vcpu;
+       struct kvm_vm *vm;
+
        if (!(kvm_check_cap(KVM_CAP_XEN_HVM) &
              KVM_XEN_HVM_CONFIG_INTERCEPT_HCALL) ) {
                print_skip("KVM_XEN_HVM_CONFIG_INTERCEPT_HCALL not available");
                exit(KSFT_SKIP);
        }
 
-       vm = vm_create_default(VCPU_ID, 0, (void *) guest_code);
-       vcpu_set_hv_cpuid(vm, VCPU_ID);
+       vm = vm_create_with_one_vcpu(&vcpu, guest_code);
+       vcpu_set_hv_cpuid(vm, vcpu->id);
 
        struct kvm_xen_hvm_config hvmc = {
                .flags = KVM_XEN_HVM_CONFIG_INTERCEPT_HCALL,
        virt_map(vm, HCALL_REGION_GPA, HCALL_REGION_GPA, 2);
 
        for (;;) {
-               volatile struct kvm_run *run = vcpu_state(vm, VCPU_ID);
+               volatile struct kvm_run *run = vcpu->run;
                struct ucall uc;
 
-               vcpu_run(vm, VCPU_ID);
+               vcpu_run(vm, vcpu->id);
 
                if (run->exit_reason == KVM_EXIT_XEN) {
                        ASSERT_EQ(run->xen.type, KVM_EXIT_XEN_HCALL);
                            run->exit_reason,
                            exit_reason_str(run->exit_reason));
 
-               switch (get_ucall(vm, VCPU_ID, &uc)) {
+               switch (get_ucall(vm, vcpu->id, &uc)) {
                case UCALL_ABORT:
                        TEST_FAIL("%s", (const char *)uc.args[0]);
                        /* NOT REACHED */