diff --git a/scripts/build-unit-test-docker b/scripts/build-unit-test-docker
index c81f965..a7eb4fb 100755
--- a/scripts/build-unit-test-docker
+++ b/scripts/build-unit-test-docker
@@ -431,7 +431,9 @@
         for line in lookup.split("\n"):
             if f"refs/heads/{branch}" in line:
                 self.pkg_def["rev"] = line.split()[0]
-            elif f"refs/heads/master" in line and "rev" not in self.pkg_def:
+            elif (
+                "refs/heads/master" in line or "refs/heads/main" in line
+            ) and "rev" not in self.pkg_def:
                 self.pkg_def["rev"] = line.split()[0]
         Package.lock.release()