diff --git a/.github/workflows/viam-build.yml b/.github/workflows/viam-build.yml
index a45875e..742efbe 100644
--- a/.github/workflows/viam-build.yml
+++ b/.github/workflows/viam-build.yml
@@ -12,8 +12,7 @@ jobs:
     - uses: viamrobotics/build-action@v1
       with:
         # note: you can replace this line with 'version: ""' if you want to test the build process without deploying
-        # version: ${{ github.ref_name }}
-        version: 2.0.1
+        version: ${{ github.ref_name }}
         ref: ${{ github.sha }}
         key-id: ${{ secrets.viam_key_id }}
         key-value: ${{ secrets.viam_key_value }}