diff --git a/.github/workflows/tests-studio.yml b/.github/workflows/tests-studio.yml
index 3d265138f..977960b7d 100644
--- a/.github/workflows/tests-studio.yml
+++ b/.github/workflows/tests-studio.yml
@@ -52,14 +52,11 @@ jobs:
           STUDIO_READ_ACCESS_TOKEN: ${{ secrets.ITERATIVE_STUDIO_READ_ACCESS_TOKEN }}
         run: |
           echo "DataChain branch: $BRANCH"
-          if [[ "$BRANCH" == "main" ]]
-          then
-              STUDIO_BRANCH=develop
-          elif git ls-remote --heads https://"$STUDIO_READ_ACCESS_TOKEN"@github.com/iterative/studio.git "$BRANCH" | grep -F "$BRANCH" 2>&1>/dev/null
+          if git ls-remote --heads https://"$STUDIO_READ_ACCESS_TOKEN"@github.com/iterative/studio.git "$BRANCH" | grep -F "$BRANCH" 2>&1>/dev/null
           then
               STUDIO_BRANCH="$BRANCH"
           else
-              STUDIO_BRANCH=develop
+              STUDIO_BRANCH=main
           fi
           echo "STUDIO_BRANCH=$STUDIO_BRANCH" >> $GITHUB_ENV
           echo "Studio branch: $STUDIO_BRANCH"