1 files changed, 2 insertions, 2 deletions

diff --git a/mm/oom_kill.c b/mm/oom_kill.c
index 578321f1c070..51c091849dcb 100644
--- a/mm/oom_kill.c
+++ b/mm/oom_kill.c
@@ -653,7 +653,7 @@ static void mark_oom_victim(struct task_struct *tsk)
 
         /* oom_mm is bound to the signal struct life time. */
         if (!cmpxchg(&tsk->signal->oom_mm, NULL, mm))
-                atomic_inc(&tsk->signal->oom_mm->mm_count);
+                mmgrab(tsk->signal->oom_mm);
 
         /*
          * Make sure that the task is woken up from uninterruptible sleep
@@ -870,7 +870,7 @@ static void oom_kill_process(struct oom_control *oc, const char *message)
 
         /* Get a reference to safely compare mm after task_unlock(victim) */
         mm = victim->mm;
-        atomic_inc(&mm->mm_count);
+        mmgrab(mm);
         /*
          * We should send SIGKILL before setting TIF_MEMDIE in order to prevent
          * the OOM victim from depleting the memory reserves from the user