1 files changed, 2 insertions, 5 deletions

diff --git a/mm/ksm.c b/mm/ksm.c
index b4d5a9deb17f..382d930a0bf1 100644
--- a/mm/ksm.c
+++ b/mm/ksm.c
@@ -1919,12 +1919,9 @@ static ssize_t run_store(struct kobject *kobj, struct kobj_attribute *attr,
         if (ksm_run != flags) {
                 ksm_run = flags;
                 if (flags & KSM_RUN_UNMERGE) {
-                        short oom_score_adj;
-
-                        oom_score_adj = test_set_oom_score_adj(OOM_SCORE_ADJ_MAX);
+                        set_current_oom_origin();
                         err = unmerge_and_remove_all_rmap_items();
-                        compare_swap_oom_score_adj(OOM_SCORE_ADJ_MAX,
-                                                                oom_score_adj);
+                        clear_current_oom_origin();
                         if (err) {
                                 ksm_run = KSM_RUN_STOP;
                                 count = err;