diff --git a/core/src/test/scala/org/apache/spark/SparkThrowableSuite.scala b/core/src/test/scala/org/apache/spark/SparkThrowableSuite.scala
index 2229f78fe2a93..946ea75686e32 100644
--- a/core/src/test/scala/org/apache/spark/SparkThrowableSuite.scala
+++ b/core/src/test/scala/org/apache/spark/SparkThrowableSuite.scala
@@ -518,7 +518,7 @@ class SparkThrowableSuite extends SparkFunSuite {
             "details" -> "implicit cast"
           ))
       },
-      errorClass = "INTERNAL_ERROR",
+      condition = "INTERNAL_ERROR",
       parameters = Map(
         "message" ->
           ("Found unused message parameters of the error class 'CANNOT_UP_CAST_DATATYPE'. " +
diff --git a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/util/DateTimeFormatterHelperSuite.scala b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/util/DateTimeFormatterHelperSuite.scala
index ba56ba19a00f8..095dc3869571d 100644
--- a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/util/DateTimeFormatterHelperSuite.scala
+++ b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/util/DateTimeFormatterHelperSuite.scala
@@ -90,7 +90,7 @@ class DateTimeFormatterHelperSuite extends SparkFunSuite {
       exception = intercept[SparkIllegalArgumentException] {
         createBuilderWithVarLengthSecondFraction(pattern)
       },
-      errorClass = "INVALID_DATETIME_PATTERN.SECONDS_FRACTION",
+      condition = "INVALID_DATETIME_PATTERN.SECONDS_FRACTION",
       parameters = Map("pattern" -> pattern))
   }
 }