diff --git a/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala b/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala
index f920a8f3d..b051dd099 100644
--- a/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala
+++ b/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala
@@ -878,6 +878,10 @@ object Utils extends Logging {
         StatusCode.HARD_SPLIT
       case 22 =>
         StatusCode.SOFT_SPLIT
+      case 23 =>
+        StatusCode.STAGE_END_TIME_OUT
+      case 24 =>
+        StatusCode.SHUFFLE_DATA_LOST
       case _ =>
         null
     }