diff --git a/doc/cp.nvim.txt b/doc/cp.nvim.txt
index 6b8097b..2f15b6e 100644
--- a/doc/cp.nvim.txt
+++ b/doc/cp.nvim.txt
@@ -422,7 +422,8 @@ Test cases use competitive programming terminology with color highlighting:
     AC      Accepted (passed) - Green
     WA      Wrong Answer (output mismatch) - Red
     TLE     Time Limit Exceeded (timeout) - Orange
-    RTE     Runtime Error (non-zero exit) - Purple
+    MLE     Memory Limit Exceeded Error (heuristic) - Orange
+    RTE     Runtime Error (other non-zero exit code) - Purple
 <
 
 ==============================================================================
diff --git a/lua/cp/runner/execute.lua b/lua/cp/runner/execute.lua
index 2f2471d..fe3667c 100644
--- a/lua/cp/runner/execute.lua
+++ b/lua/cp/runner/execute.lua
@@ -141,7 +141,7 @@ function M.run(cmd, stdin, timeout_ms, memory_mb)
   end
 
   local mled = false
-  if peak_mb >= memory_mb or (code ~= 0 and not tled) and (near_cap or oom_hint) or code == 137 then
+  if peak_mb >= memory_mb or near_cap or oom_hint then
     mled = true
   end