{
        struct kvm_sregs sregs;
 
+       TEST_ASSERT_EQ(vm->mode, VM_MODE_PXXV48_4K);
+
        /* Set mode specific system register values. */
        vcpu_sregs_get(vcpu, &sregs);
 
 
        kvm_setup_gdt(vm, &sregs.gdt);
 
-       switch (vm->mode) {
-       case VM_MODE_PXXV48_4K:
-               sregs.cr0 = X86_CR0_PE | X86_CR0_NE | X86_CR0_PG;
-               sregs.cr4 |= X86_CR4_PAE | X86_CR4_OSFXSR;
-               sregs.efer |= (EFER_LME | EFER_LMA | EFER_NX);
-
-               kvm_seg_set_unusable(&sregs.ldt);
-               kvm_seg_set_kernel_code_64bit(vm, DEFAULT_CODE_SELECTOR, &sregs.cs);
-               kvm_seg_set_kernel_data_64bit(vm, DEFAULT_DATA_SELECTOR, &sregs.ds);
-               kvm_seg_set_kernel_data_64bit(vm, DEFAULT_DATA_SELECTOR, &sregs.es);
-               kvm_setup_tss_64bit(vm, &sregs.tr, 0x18);
-               break;
-
-       default:
-               TEST_FAIL("Unknown guest mode, mode: 0x%x", vm->mode);
-       }
+       sregs.cr0 = X86_CR0_PE | X86_CR0_NE | X86_CR0_PG;
+       sregs.cr4 |= X86_CR4_PAE | X86_CR4_OSFXSR;
+       sregs.efer |= (EFER_LME | EFER_LMA | EFER_NX);
+
+       kvm_seg_set_unusable(&sregs.ldt);
+       kvm_seg_set_kernel_code_64bit(vm, DEFAULT_CODE_SELECTOR, &sregs.cs);
+       kvm_seg_set_kernel_data_64bit(vm, DEFAULT_DATA_SELECTOR, &sregs.ds);
+       kvm_seg_set_kernel_data_64bit(vm, DEFAULT_DATA_SELECTOR, &sregs.es);
+       kvm_setup_tss_64bit(vm, &sregs.tr, 0x18);
 
        sregs.cr3 = vm->pgd;
        vcpu_sregs_set(vcpu, &sregs);