diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
index 407e5b1..81eb2ee 100644
--- a/.github/workflows/push.yml
+++ b/.github/workflows/push.yml
@@ -15,7 +15,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        model: [yolo11n,yolov8s-worldv2,yoloe11s]
+        model: [yolo11n, yolov8s-worldv2, yoloe11s]
     steps:
       - name: Checkout
         uses: actions/checkout@v4
diff --git a/yolo11n/cog.yaml b/yolo11n/cog.yaml
index 8b5fd54..caec2c9 100644
--- a/yolo11n/cog.yaml
+++ b/yolo11n/cog.yaml
@@ -4,7 +4,6 @@ build:
   gpu: false
   python_version: "3.12"
   system_packages:
-    - "libgl1-mesa-glx"
     - "libglib2.0-0"
     - "ffmpeg"
   python_packages:
diff --git a/yoloe11s/cog.yaml b/yoloe11s/cog.yaml
index 38d4dc6..beda2c8 100644
--- a/yoloe11s/cog.yaml
+++ b/yoloe11s/cog.yaml
@@ -4,7 +4,6 @@ build:
   gpu: false
   python_version: "3.12"
   system_packages:
-    - "libgl1-mesa-glx"
     - "libglib2.0-0"
     - "ffmpeg"
   python_packages:
diff --git a/yoloe11s/predict.py b/yoloe11s/predict.py
index 20df8d2..85c2532 100644
--- a/yoloe11s/predict.py
+++ b/yoloe11s/predict.py
@@ -22,13 +22,14 @@ def setup(self) -> None:
 
     def re_init_model(self, class_names: str) -> None:
         """Re-Initialize model with class names."""
-        if class_names.strip() != "":
+        if not class_names.strip():
+            # Load YOLOE-11s model prompt free model into memory
+            self.model = YOLOE("yoloe-11s-seg-pf.pt")
+
+        else:
             self.model = YOLOE("yoloe-11s-seg.pt")
             class_list = class_names.split(", ")
             self.model.set_classes(class_list, self.model.get_text_pe(class_list))
-        else:
-            # Load YOLOE-11s model prompt free model into memory
-            self.model = YOLOE("yoloe-11s-seg-pf.pt")
 
     def predict(
         self,
diff --git a/yolov8s-worldv2/cog.yaml b/yolov8s-worldv2/cog.yaml
index f01edbf..41a67a4 100644
--- a/yolov8s-worldv2/cog.yaml
+++ b/yolov8s-worldv2/cog.yaml
@@ -4,7 +4,6 @@ build:
   gpu: false
   python_version: "3.12"
   system_packages:
-    - "libgl1-mesa-glx"
     - "libglib2.0-0"
     - "ffmpeg"
   python_packages: