diff --git a/CITATION.cff b/CITATION.cff
index 30a91d0..7912b69 100644
--- a/CITATION.cff
+++ b/CITATION.cff
@@ -8,7 +8,7 @@ title: "tf-keras-vis"
 repository: "https://github.com/keisen/tf-keras-vis"
 url: "https://keisen.github.io/tf-keras-vis-docs/"
 type: software
-version: 0.8.0
+version: 0.8.1
 date-released: "2021-08-14"
 license-url: "https://github.com/keisen/tf-keras-vis/blob/master/LICENSE"
 references:
diff --git a/VERSION b/VERSION
index 8adc70f..c18d72b 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-0.8.0
\ No newline at end of file
+0.8.1
\ No newline at end of file