diff --git a/doc/source/conf.py b/doc/source/conf.py
index 1d4fd1d..ecc17c1 100644
--- a/doc/source/conf.py
+++ b/doc/source/conf.py
@@ -28,9 +28,9 @@
 author = 'Yahoo Inc'
 
 # The short X.Y version
-version = '2.2.4'
+version = '2.2.5'
 # The full version, including alpha/beta/rc tags
-release = '2.2.4'
+release = '2.2.5'
 
 
 # -- General configuration ---------------------------------------------------
diff --git a/pom.xml b/pom.xml
index 2aaee41..c6352ea 100644
--- a/pom.xml
+++ b/pom.xml
@@ -5,7 +5,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.yahoo.ml</groupId>
     <artifactId>tensorflowonspark</artifactId>
-    <version>2.2.4-SNAPSHOT</version>
+    <version>2.2.5-SNAPSHOT</version>
     <packaging>jar</packaging>
     <name>tensorflowonspark</name>
     <description>Spark Scala inferencing for TensorFlowOnSpark</description>
diff --git a/setup.cfg b/setup.cfg
index 5f10c54..6bcfab4 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -9,16 +9,16 @@ classifiers =
     License :: OSI Approved :: Apache Software License
     Topic :: Software Development :: Libraries
     Programming Language :: Python :: 3 :: Only
-    Programming Language :: Python :: 3.6
     Programming Language :: Python :: 3.7
     Programming Language :: Python :: 3.8
+    Programming Language :: Python :: 3.9
 description = Deep learning with TensorFlow on Apache Spark clusters
 license = Apache 2.0
 long_description = file:README.md
 long_description_content_type = text/markdown
 name = tensorflowonspark
 url = https://github.com/yahoo/TensorFlowOnSpark
-version = 2.2.4
+version = 2.2.5
 
 [options]
 packages =
diff --git a/tensorflowonspark/__init__.py b/tensorflowonspark/__init__.py
index 21ee7c4..3de2eb8 100644
--- a/tensorflowonspark/__init__.py
+++ b/tensorflowonspark/__init__.py
@@ -2,4 +2,4 @@
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s (%(threadName)s-%(process)d) %(message)s")
 
-__version__ = "2.2.4"
+__version__ = "2.2.5"