1 files changed, 3 insertions, 3 deletions

diff --git a/arch/x86/kernel/vm86_32.c b/arch/x86/kernel/vm86_32.c
index 54abcc0baf23..5c9687b1bde6 100644
--- a/arch/x86/kernel/vm86_32.c
+++ b/arch/x86/kernel/vm86_32.c
@@ -561,9 +561,9 @@ int handle_vm86_trap(struct kernel_vm86_regs *regs, long error_code, int trapno)
                 if ((trapno == 3) || (trapno == 1)) {
                         KVM86->regs32->ax = VM86_TRAP + (trapno << 8);
                         /* setting this flag forces the code in entry_32.S to
-                           call save_v86_state() and change the stack pointer
-                           to KVM86->regs32 */
-                        set_thread_flag(TIF_IRET);
+                           the path where we call save_v86_state() and change
+                           the stack pointer to KVM86->regs32 */
+                        set_thread_flag(TIF_NOTIFY_RESUME);
                         return 0;
                 }
                 do_int(regs, trapno, (unsigned char __user *) (regs->pt.ss << 4), SP(regs));