__activate_vm(vcpu->kvm);
 
        __vgic_restore_state(vcpu);
-       __timer_enable_traps(vcpu);
 
        /*
         * We must restore the 32-bit state before the sysregs, thanks
 
        __sysreg_save_guest_state(guest_ctxt);
        __sysreg32_save_state(vcpu);
-       __timer_disable_traps(vcpu);
        __vgic_save_state(vcpu);
 
        __deactivate_traps(vcpu);
 
        write_sysreg(cntvoff, cntvoff_el2);
 }
 
+/*
+ * Should only be called on non-VHE systems.
+ * VHE systems use EL2 timers and configure EL1 timers in kvm_timer_init_vhe().
+ */
 void __hyp_text __timer_disable_traps(struct kvm_vcpu *vcpu)
 {
-       /*
-        * We don't need to do this for VHE since the host kernel runs in EL2
-        * with HCR_EL2.TGE ==1, which makes those bits have no impact.
-        */
-       if (!has_vhe()) {
-               u64 val;
+       u64 val;
 
-               /* Allow physical timer/counter access for the host */
-               val = read_sysreg(cnthctl_el2);
-               val |= CNTHCTL_EL1PCTEN | CNTHCTL_EL1PCEN;
-               write_sysreg(val, cnthctl_el2);
-       }
+       /* Allow physical timer/counter access for the host */
+       val = read_sysreg(cnthctl_el2);
+       val |= CNTHCTL_EL1PCTEN | CNTHCTL_EL1PCEN;
+       write_sysreg(val, cnthctl_el2);
 }
 
+/*
+ * Should only be called on non-VHE systems.
+ * VHE systems use EL2 timers and configure EL1 timers in kvm_timer_init_vhe().
+ */
 void __hyp_text __timer_enable_traps(struct kvm_vcpu *vcpu)
 {
-       if (!has_vhe()) {
-               u64 val;
+       u64 val;
 
-               /*
-                * Disallow physical timer access for the guest
-                * Physical counter access is allowed
-                */
-               val = read_sysreg(cnthctl_el2);
-               val &= ~CNTHCTL_EL1PCEN;
-               val |= CNTHCTL_EL1PCTEN;
-               write_sysreg(val, cnthctl_el2);
-       }
+       /*
+        * Disallow physical timer access for the guest
+        * Physical counter access is allowed
+        */
+       val = read_sysreg(cnthctl_el2);
+       val &= ~CNTHCTL_EL1PCEN;
+       val |= CNTHCTL_EL1PCTEN;
+       write_sysreg(val, cnthctl_el2);
 }