diff --git a/configs/mcli/mitchish-instruct.yml b/configs/mcli/mitchish-instruct.yml
index 84fa7b57d..59ee0c270 100644
--- a/configs/mcli/mitchish-instruct.yml
+++ b/configs/mcli/mitchish-instruct.yml
@@ -8,7 +8,7 @@ compute:
 integrations:
   - integration_type: git_repo
     git_repo: allenai/LLM
-    git_branch: epwalsh/fine-tune-with-label-masking
+    git_branch: main
     pip_install: -e .
     ssh_clone: true
 command: |-