diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 3c642c74..b64d6581 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -12,7 +12,7 @@ on:
 
 jobs:
   test:
-    uses: ocadotechnology/codeforlife-workspace/.github/workflows/test-python-code.yaml@workspace-145 # TODO: set to @main
+    uses: ocadotechnology/codeforlife-workspace/.github/workflows/test-python-code.yaml@main
     secrets: inherit
     with:
       # Cannot be set with an env var. Value must match in the release job.