diff --git a/pyproject.toml b/pyproject.toml
index c68feee..7afb358 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,7 +18,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "flagsparse"
-version = "0.4.0"
+version = "0.4.0-dev1"
 description = "FlagSparse - GPU sparse operations package"
 readme = "README.md"
 requires-python = ">=3.10"
diff --git a/setup.py b/setup.py
index 84a88dd..1e11027 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@
 
 setup(
     name="flagsparse",
-    version="0.4.0",
+    version="0.4.0-dev1",
     description="FlagSparse - GPU sparse operations package",
     package_dir={"": "src"},
     packages=find_packages(where="src"),
diff --git a/src/flagsparse/__init__.py b/src/flagsparse/__init__.py
index bdf96b2..cc1d276 100644
--- a/src/flagsparse/__init__.py
+++ b/src/flagsparse/__init__.py
@@ -14,7 +14,11 @@
 
 """FlagSparse package."""
 
-__version__ = "0.4.0"
+from importlib.metadata import version, PackageNotFoundError
+try:
+    __version__ = version("flagsparse")
+except PackageNotFoundError:
+    __version__ = "0.0.0"
 
 __all__ = [
     "flagsparse_gather",