diff --git a/flint-spark-integration/src/test/scala/org/apache/spark/sql/FlintCatalogSuite.scala b/flint-spark-integration/src/test/scala/org/apache/spark/sql/FlintCatalogSuite.scala
index f0aea2641..3c75cf541 100644
--- a/flint-spark-integration/src/test/scala/org/apache/spark/sql/FlintCatalogSuite.scala
+++ b/flint-spark-integration/src/test/scala/org/apache/spark/sql/FlintCatalogSuite.scala
@@ -83,7 +83,7 @@ class FlintCatalogSuite extends SparkFunSuite with MockitoSugar {
     }
 
     def shouldThrowException(): Unit = {
-      assertThrows[RuntimeException] {
+      assertThrows[IllegalStateException] {
         resolveCatalogName(spark, catalog)
       }
     }