diff --git a/coral-spark/src/main/java/com/linkedin/coral/spark/transformers/TransportUDFTransformer.java b/coral-spark/src/main/java/com/linkedin/coral/spark/transformers/TransportUDFTransformer.java
index 54b2be17d..baa5f2c85 100644
--- a/coral-spark/src/main/java/com/linkedin/coral/spark/transformers/TransportUDFTransformer.java
+++ b/coral-spark/src/main/java/com/linkedin/coral/spark/transformers/TransportUDFTransformer.java
@@ -1,5 +1,5 @@
 /**
- * Copyright 2018-2024 LinkedIn Corporation. All rights reserved.
+ * Copyright 2018-2025 LinkedIn Corporation. All rights reserved.
  * Licensed under the BSD-2 Clause license.
  * See LICENSE in the project root for license information.
  */
@@ -88,7 +88,7 @@ public ScalaVersion getScalaVersionOfSpark() {
       String sparkVersion = SparkSession.active().version();
       if (sparkVersion.matches("2\\.[\\d\\.]*")) {
         return ScalaVersion.SCALA_2_11;
-      } else if (sparkVersion.matches("3\\.[\\d\\.]*")) {
+      } else if (sparkVersion.matches("3\\d*\\.[\\d\\.]*")) {
         return ScalaVersion.SCALA_2_12;
       } else {
         throw new IllegalStateException(String.format("Unsupported Spark Version %s", sparkVersion));