diff --git a/.bumpversion.cfg b/.bumpversion.cfg
index f12c0a3..6d50e9b 100644
--- a/.bumpversion.cfg
+++ b/.bumpversion.cfg
@@ -1,8 +1,10 @@
 [bumpversion]
-current_version = 0.3.0
+current_version = 0.3.1
 commit = True
 tag = False
 
 [bumpversion:file:setup.py]
 
+[bumpversion:file:CITATION.cff]
+
 [bumpversion:file:tf_explain/__init__.py]
diff --git a/CITATION.cff b/CITATION.cff
index a28c980..60a8092 100644
--- a/CITATION.cff
+++ b/CITATION.cff
@@ -4,7 +4,7 @@
 cff-version: 1.2.0
 title: tf-explain
 abstract: Interpretability Methods for tf.keras models with TensorFlow 2.x
-version: 0.3.0
+version: 0.3.1
 date-released: 2021-02-04
 message: "If you use tf-explain in your research, please cite it using these metadata."
 type: software
diff --git a/setup.py b/setup.py
index 7ead67c..155d27b 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name="tf-explain",
-    version="0.3.0",
+    version="0.3.1",
     description="Interpretability Callbacks for Tensorflow 2.0",
     long_description=long_description,
     long_description_content_type="text/markdown",
diff --git a/tf_explain/__init__.py b/tf_explain/__init__.py
index a420a6b..1eb0fa4 100644
--- a/tf_explain/__init__.py
+++ b/tf_explain/__init__.py
@@ -5,7 +5,7 @@
 callbacks to ease neural network's understanding.
 """
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 try:
     import cv2