diff --git a/docs/source/conf.py b/docs/source/conf.py
index e4c97071..782a1827 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -28,9 +28,9 @@
 author = 'Yahoo Inc'
 
 # The short X.Y version
-version = '2.1.0'
+version = '2.1.1'
 # The full version, including alpha/beta/rc tags
-release = '2.1.0'
+release = '2.1.1'
 
 
 # -- General configuration ---------------------------------------------------
diff --git a/pom.xml b/pom.xml
index 73594309..1e75b16e 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.1.0-SNAPSHOT</version>
+    <version>2.1.1-SNAPSHOT</version>
     <packaging>jar</packaging>
     <name>tensorflowonspark</name>
     <description>Spark Scala inferencing for TensorFlowOnSpark</description>
diff --git a/setup.py b/setup.py
index c11d32c3..bfbe7e76 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 setup(
   name='tensorflowonspark',
   packages=['tensorflowonspark'],
-  version='2.1.0',
+  version='2.1.1',
   description='Deep learning with TensorFlow on Apache Spark clusters',
   long_description=long_description,
   long_description_content_type='text/markdown',
diff --git a/tensorflowonspark/__init__.py b/tensorflowonspark/__init__.py
index 044335d1..8b086af5 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.1.0"
+__version__ = "2.1.1"