diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 8419e077..1e8bc6be 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -6,7 +6,7 @@ on:
 jobs:
   pypi-publish:
     name: Publish release to PyPI
-    runs-on: ubuntu-22.04
+    runs-on: ubuntu-latest
     environment:
       name: pypi
       url: https://pypi.org/p/optillm
@@ -30,7 +30,7 @@ jobs:
 
   docker-publish:
     name: Publish Docker image
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     needs: pypi-publish
     permissions:
       contents: read
diff --git a/setup.py b/setup.py
index 1d7285f7..29424ab8 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="optillm",
-    version="0.0.26",
+    version="0.0.27",
     packages=find_packages(),
     py_modules=['optillm'],
     package_data={