1 files changed, 4 insertions, 2 deletions

diff --git a/kernel/context_tracking.c b/kernel/context_tracking.c
index 942835c12ae5..1f47119c5b09 100644
--- a/kernel/context_tracking.c
+++ b/kernel/context_tracking.c
@@ -141,12 +141,13 @@ void user_exit(void)
         local_irq_restore(flags);
 }
 
+#ifdef CONFIG_VIRT_CPU_ACCOUNTING_GEN
 void guest_enter(void)
 {
         if (vtime_accounting_enabled())
                 vtime_guest_enter(current);
         else
-                __guest_enter();
+                current->flags |= PF_VCPU;
 }
 EXPORT_SYMBOL_GPL(guest_enter);
 
@@ -155,9 +156,10 @@ void guest_exit(void)
         if (vtime_accounting_enabled())
                 vtime_guest_exit(current);
         else
-                __guest_exit();
+                current->flags &= ~PF_VCPU;
 }
 EXPORT_SYMBOL_GPL(guest_exit);
+#endif /* CONFIG_VIRT_CPU_ACCOUNTING_GEN */
 
 
 /**