diff --git a/.github/workflows/pypi.yml b/.github/workflows/pypi.yml
index 0b94f6f3..7cd9c3ec 100644
--- a/.github/workflows/pypi.yml
+++ b/.github/workflows/pypi.yml
@@ -15,12 +15,12 @@ jobs:
         python-version: '3.9'
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip uv
-        python -m uv pip install setuptools wheel twine
+        pip install --upgrade pip uv
+        uv pip install build twine
     - name: Build and publish
       env:
         TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
         TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
       run: |
-        python setup.py sdist bdist_wheel
+        python -m build
         twine upload dist/*
diff --git a/segmentation_models_pytorch/__version__.py b/segmentation_models_pytorch/__version__.py
index 478f3a06..b87975ee 100644
--- a/segmentation_models_pytorch/__version__.py
+++ b/segmentation_models_pytorch/__version__.py
@@ -1,3 +1,3 @@
-VERSION = (0, 3, "5dev0")
+VERSION = (0, 4, 0)
 
 __version__ = ".".join(map(str, VERSION))