diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index f2a40b9..88e0e86 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -16,7 +16,7 @@ jobs:
       - uses: actions/checkout@v4
         with:
           token: ${{ secrets.PAT_ACTIONS }}
-          ref: main
+          ref: v1
           fetch-depth: 0
       - uses: actions/setup-python@v5
       - name: Install dependencies
@@ -42,7 +42,7 @@ jobs:
           git add setup.py
           git add skyflow/version.py
           git commit -m "[AUTOMATED] Public Release - ${{ steps.previoustag.outputs.tag }}"
-          git push origin
+          git push origin HEAD:v1
           
       - name: Build and publish
         env:
diff --git a/setup.py b/setup.py
index f65126a..98f579a 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 
 if sys.version_info < (3, 7):
     raise RuntimeError("skyflow requires Python 3.7+")
-current_version = '1.16.1.dev0+eec539e'
+current_version = '2.1.1.dev0+67ee0bb'
 
 setup(
     name='skyflow',
diff --git a/skyflow/version.py b/skyflow/version.py
index 3810172..91d11a0 100644
--- a/skyflow/version.py
+++ b/skyflow/version.py
@@ -1 +1 @@
-SDK_VERSION = '1.16.1'
\ No newline at end of file
+SDK_VERSION = '2.1.1.dev0+67ee0bb'
\ No newline at end of file