diff --git a/spark-sql-application/src/test/scala/org/apache/spark/sql/FlintREPLTest.scala b/spark-sql-application/src/test/scala/org/apache/spark/sql/FlintREPLTest.scala
index 72aab2b30..aa3b929db 100644
--- a/spark-sql-application/src/test/scala/org/apache/spark/sql/FlintREPLTest.scala
+++ b/spark-sql-application/src/test/scala/org/apache/spark/sql/FlintREPLTest.scala
@@ -465,7 +465,7 @@ class FlintREPLTest
   }
 
   test("handleGeneralException should handle MetaException with AccessDeniedException properly") {
-    val mockFlintCommand = mock[FlintStatement]
+    val mockFlintCommand = mock[FlintCommand]
 
     // Simulate the root cause being MetaException
     val exception = new org.apache.hadoop.hive.metastore.api.MetaException(