diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 2da9f54..2a31797 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -27,5 +27,6 @@ jobs:
         TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
       run: |
         sed -i "s|{{VERSION}}|${{ github.event.release.tag_name }}|g" ./setup.py
+        sed -i "s|{{VERSION}}|${{ github.event.release.tag_name }}|g" ./extra_keras_datasets/__init__.py
         python setup.py sdist bdist_wheel
         twine upload dist/*
\ No newline at end of file
diff --git a/extra_keras_datasets/__init__.py b/extra_keras_datasets/__init__.py
index 561e3d7..11a43e5 100644
--- a/extra_keras_datasets/__init__.py
+++ b/extra_keras_datasets/__init__.py
@@ -1,6 +1,6 @@
 from __future__ import absolute_import
 
-__version__ = '0.1.5'
+__version__ = '{{VERSION}}'
 
 from . import emnist
 from . import kmnist