diff --git a/include/linux/oom.h b/include/linux/oom.h
index 9ea3ccb18c6f..3f40397c6860 100644
--- a/include/linux/oom.h
+++ b/include/linux/oom.h
@@ -64,7 +64,11 @@ static inline bool oom_task_origin(const struct task_struct *p)
 
 static inline bool tsk_is_oom_victim(struct task_struct * tsk)
 {
+#ifdef CONFIG_ANDROID_SIMPLE_LMK
+	return test_ti_thread_flag(task_thread_info(tsk), TIF_MEMDIE);
+#else
 	return tsk->signal->oom_mm;
+#endif
 }
 
 /*