diff --git a/setup.py b/setup.py
index 1678bbd..055673e 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@ from setuptools import find_packages
 from setuptools import setup
 
 setup(name='tensorflow_privacy',
-      version='0.2.0',
+      version='0.2.1',
       url='https://github.com/tensorflow/privacy',
       license='Apache-2.0',
       install_requires=[