diff --git a/tensorflow/java/maven/spark-tensorflow-connector/pom.xml b/tensorflow/java/maven/spark-tensorflow-connector/pom.xml
index 7bc2f8e1841e61..36ea7c4addd4eb 100644
--- a/tensorflow/java/maven/spark-tensorflow-connector/pom.xml
+++ b/tensorflow/java/maven/spark-tensorflow-connector/pom.xml
@@ -34,7 +34,7 @@
         <maven.compiler.version>3.0</maven.compiler.version>
         <java.version>1.8</java.version>
         <spark.version>2.3.1</spark.version>
-        <yarn.api.version>2.7.3</yarn.api.version>
+        <yarn.api.version>3.3.1</yarn.api.version>
         <junit.version>4.11</junit.version>
     </properties>