diff --git a/.github/workflows/gar-publish-dev.yaml b/.github/workflows/gar-publish-dev.yaml
index b5a7408..9b0bb31 100644
--- a/.github/workflows/gar-publish-dev.yaml
+++ b/.github/workflows/gar-publish-dev.yaml
@@ -15,9 +15,7 @@ jobs:
 
     steps:
       - name: 'Checkout GitHub Action'
-        uses: actions/checkout@main
-        with:
-          ref: dev  # Checkout the dev branch
+        uses: actions/checkout@v4
         
       - id: 'auth'
         name: 'Authenticate to Google Cloud'
@@ -36,5 +34,5 @@ jobs:
       - name: 'Build Inventory Image'
         working-directory: .
         run: |
-          docker build --build-arg GITHUB_TOKEN=${{secrets.GHCR_PAT}} --build-arg BRANCH=dev . --tag $GAR_LOCATION-docker.pkg.dev/$PROJECT_ID/internal/datadreamer:dev
+          docker build --build-arg GITHUB_TOKEN=${{secrets.GHCR_PAT}} --build-arg BRANCH=${{ github.ref_name }} . --tag $GAR_LOCATION-docker.pkg.dev/$PROJECT_ID/internal/datadreamer:dev
           docker push $GAR_LOCATION-docker.pkg.dev/$PROJECT_ID/internal/datadreamer --all-tags
\ No newline at end of file
diff --git a/datadreamer/utils/luxonis_dataset_converter.py b/datadreamer/utils/luxonis_dataset_converter.py
index 5cde762..1173994 100644
--- a/datadreamer/utils/luxonis_dataset_converter.py
+++ b/datadreamer/utils/luxonis_dataset_converter.py
@@ -80,6 +80,12 @@ def dataset_generator():
                 image_full_path = os.path.join(dataset_dir, image_path)
                 width, height = Image.open(image_full_path).size
                 labels = data[image_path]["labels"]
+
+                if len(labels) == 0:
+                    yield {
+                        "file": image_full_path,
+                    }
+
                 for label in labels:
                     yield {
                         "file": image_full_path,
diff --git a/media/coverage_badge.svg b/media/coverage_badge.svg
index 6c15cac..179c6a1 100644
--- a/media/coverage_badge.svg
+++ b/media/coverage_badge.svg
@@ -9,13 +9,13 @@
     </mask>
     <g mask="url(#a)">
         <path fill="#555" d="M0 0h63v20H0z"/>
-        <path fill="#a4a61d" d="M63 0h36v20H63z"/>
+        <path fill="#dfb317" d="M63 0h36v20H63z"/>
         <path fill="url(#b)" d="M0 0h99v20H0z"/>
     </g>
     <g fill="#fff" text-anchor="middle" font-family="DejaVu Sans,Verdana,Geneva,sans-serif" font-size="11">
         <text x="31.5" y="15" fill="#010101" fill-opacity=".3">coverage</text>
         <text x="31.5" y="14">coverage</text>
-        <text x="80" y="15" fill="#010101" fill-opacity=".3">75%</text>
-        <text x="80" y="14">75%</text>
+        <text x="80" y="15" fill="#010101" fill-opacity=".3">63%</text>
+        <text x="80" y="14">63%</text>
     </g>
 </svg>
diff --git a/requirements.txt b/requirements.txt
index 2ab6b13..df5cc3e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,6 +12,6 @@ accelerate>=0.25.0
 scipy>=1.10.0
 bitsandbytes>=0.42.0
 nltk>=3.8.1
-luxonis-ml[all]>=0.4.1
+luxonis-ml[all]>=0.5.0
 python-box>=7.1.1
 gcsfs>=2023.1.0
\ No newline at end of file