diff --git a/.github/workflows/pypi.yml b/.github/workflows/pypi.yml
index e18dd6d..29a844d 100644
--- a/.github/workflows/pypi.yml
+++ b/.github/workflows/pypi.yml
@@ -28,11 +28,11 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
-          pip install build
-      - name: Build package
-        run: python -m build
-      - name: Publish package
-        uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
-        with:
-          user: __token__
-          password: ${{ secrets.PYPI_TOKEN }}
+          pip install setuptools wheel twine
+      - name: Build and publish
+        env:
+          TWINE_USERNAME: __token__
+          TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}
+        run: |
+          python setup.py bdist_wheel sdist
+          twine upload dist/*.whl dist/*.tar.gz
diff --git a/setup.py b/setup.py
index dca8fb1..b70ff3f 100644
--- a/setup.py
+++ b/setup.py
@@ -17,6 +17,7 @@ setup(
         "pandas",
         "ipython",
         "tensorflow<=2.11.1",
+        "keras"
     ],
     packages=find_packages(exclude=[]),
     keywords=["pso", "tensorflow", "keras"],