From 3a3ba162b42e0ad7e9fdced95b5c30d966ae240f Mon Sep 17 00:00:00 2001
From: Joao Amaral <7281460+joaopamaral@users.noreply.github.com>
Date: Mon, 7 Oct 2024 20:22:39 -0300
Subject: [PATCH] Fix

---
 .../org/apache/kyuubi/operation/BatchJobSubmission.scala      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/kyuubi-server/src/main/scala/org/apache/kyuubi/operation/BatchJobSubmission.scala b/kyuubi-server/src/main/scala/org/apache/kyuubi/operation/BatchJobSubmission.scala
index 39098a58698..590ddaebafb 100644
--- a/kyuubi-server/src/main/scala/org/apache/kyuubi/operation/BatchJobSubmission.scala
+++ b/kyuubi-server/src/main/scala/org/apache/kyuubi/operation/BatchJobSubmission.scala
@@ -161,8 +161,8 @@ class BatchJobSubmission(
     }
   }
 
-  private def getAppState(operState: OperationState, appState: ApplicationState): Unit = {
-    if (state == ERROR && appState != ApplicationState.FAILED) {
+  private def getAppState(operState: OperationState, appState: ApplicationState.ApplicationState): ApplicationState.ApplicationState = {
+    if (state == OperationState.ERROR && appState != ApplicationState.FAILED) {
       ApplicationState.UNKNOWN
     }
     appState