diff --git a/.github/workflows/publish-to-pypi.yaml b/.github/workflows/publish-to-pypi.yaml
index 33d84c3..d83ec5e 100644
--- a/.github/workflows/publish-to-pypi.yaml
+++ b/.github/workflows/publish-to-pypi.yaml
@@ -2,11 +2,6 @@ name: build
 on:
   release:
     types: [published]
-  # workflow_run:
-  #   workflows: ["tests"]
-  #   branches: [master]
-  #   types:
-  #     - completed
 jobs:
   build-and-publish:
     name: Build and publish Python 🐍 distributions 📦 to PyPI
@@ -38,13 +33,7 @@ jobs:
         --wheel
         --outdir dist/
         .
-    # - name: Publish distribution 📦 to Test PyPI
-    #   uses: pypa/gh-action-pypi-publish@master
-    #   with:
-    #     password: ${{ secrets.TEST_PYPI_API_TOKEN }}
-    #     repository_url: https://test.pypi.org/legacy/
     - name: Publish distribution 📦 to PyPI
-      # if: startsWith(github.ref, 'refs/tags')
       uses: pypa/gh-action-pypi-publish@master
       with:
         password: ${{ secrets.PYPI_API_TOKEN }}
\ No newline at end of file
diff --git a/README.md b/README.md
index 057c577..638609b 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 # tbparse
 
 [![tests](https://github.com/j3soon/tbparse/actions/workflows/test-with-tox.yaml/badge.svg?branch=master&event=push)](https://github.com/j3soon/tbparse/actions/workflows/test-with-tox.yaml)
-[![build](https://github.com/j3soon/tbparse/actions/workflows/publish-to-pypi.yaml/badge.svg?branch=master&event=push)](https://github.com/j3soon/tbparse/actions/workflows/publish-to-pypi.yaml)
+[![build](https://github.com/j3soon/tbparse/actions/workflows/publish-to-pypi.yaml/badge.svg)](https://github.com/j3soon/tbparse/actions/workflows/publish-to-pypi.yaml)
 
 Read, parse, and plot tensorboard event logs with ease!
 
diff --git a/docs/pages/changelog.rst b/docs/pages/changelog.rst
index 9740b21..d85cea2 100644
--- a/docs/pages/changelog.rst
+++ b/docs/pages/changelog.rst
@@ -2,6 +2,13 @@
 Change Log
 ===================================
 
+v0.0.2
+===================================
+
+Fixes:
+
+* Fix PyPI release issue
+
 v0.0.1
 ===================================
 
diff --git a/setup.py b/setup.py
index 61f3c75..4a43342 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="tbparse",
-    version="0.0.1",
+    version="0.0.2",
     author="Johnson",
     author_email="j3.soon@msa.hinet.net",
     description="A simple parser for reading tensorboard logs",
@@ -20,8 +20,8 @@
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
-    package_dir={"": "tbparse"},
-    packages=setuptools.find_packages(where="tbparse"),
+    package_dir={"": "."},
+    packages=setuptools.find_packages(where=".", exclude=("tests*",)),
     python_requires=">=3.7",
     install_requires=[
         'pandas>=1.3.0',