diff --git a/kyuubi-server/src/main/scala/org/apache/kyuubi/engine/YarnApplicationOperation.scala b/kyuubi-server/src/main/scala/org/apache/kyuubi/engine/YarnApplicationOperation.scala
index 1f672ad701e..18d0006a592 100644
--- a/kyuubi-server/src/main/scala/org/apache/kyuubi/engine/YarnApplicationOperation.scala
+++ b/kyuubi-server/src/main/scala/org/apache/kyuubi/engine/YarnApplicationOperation.scala
@@ -184,7 +184,8 @@ object YarnApplicationOperation extends Logging {
         ApplicationState.RUNNING
       case (YarnApplicationState.FINISHED, FinalApplicationStatus.SUCCEEDED) =>
         ApplicationState.FINISHED
-      case (YarnApplicationState.FAILED, FinalApplicationStatus.FAILED) =>
+      case (YarnApplicationState.FINISHED, FinalApplicationStatus.FAILED) |
+          (YarnApplicationState.FAILED, FinalApplicationStatus.FAILED) =>
         ApplicationState.FAILED
       case (YarnApplicationState.KILLED, FinalApplicationStatus.KILLED) =>
         ApplicationState.KILLED