diff --git a/.github/workflows/pypi.yml b/.github/workflows/pypi.yml
index 25d1e2d..f9a7ba1 100644
--- a/.github/workflows/pypi.yml
+++ b/.github/workflows/pypi.yml
@@ -46,7 +46,7 @@ jobs:
           pip install -U setuptools
           pip install -r requirements.txt
           python3 -m pip install --no-deps --pre --index-url https://test.pypi.org/simple/ scraibe>=0.1.3
-
+          python3 -m -c "import scraibe; print(scraibe.__version__)"
 
   # publish-to-pypi:
   #   name: Conditional Publish to PyPI or Dev Repository
diff --git a/.github/workflows/pytest.yaml b/.github/workflows/pytest.yaml
index 973571f..a1b0234 100644
--- a/.github/workflows/pytest.yaml
+++ b/.github/workflows/pytest.yaml
@@ -1,8 +1,6 @@
 name: Run Tests
 
 on:
- #push:
-  
   pull_request:
     branches: ['main', 'develop']
   workflow_dispatch:
@@ -30,10 +28,7 @@ jobs:
              sudo apt-get install libsndfile1-dev 
              sudo apt-get install ffmpeg 
              pip install pytest
-
-
-           
-
+             
       - name: Run pytest
         env: 
           HF_TOKEN : ${{ secrets.HF_TOKEN }}
diff --git a/scraibe/__init__.py b/scraibe/__init__.py
index 9f93efc..4338879 100644
--- a/scraibe/__init__.py
+++ b/scraibe/__init__.py
@@ -7,7 +7,6 @@ from .diarisation import *
 from .misc import *
 
 from .cli import *
- 
- # set __version__ attribute
- 
-__version__ = "0.0.0"
+
+from ._version import __version__ 
+
diff --git a/scraibe/_version.py b/scraibe/_version.py
index 3ab50aa..3aa0d7b 100644
--- a/scraibe/_version.py
+++ b/scraibe/_version.py
@@ -1,3 +1 @@
- # set __version__ attribute
- 
 __version__ = "0.0.0"
\ No newline at end of file