diff --git a/docs/conf.py b/docs/conf.py
index fab3660b..1d72caf9 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -176,7 +176,7 @@ def linkcode_resolve(domain, info):
 
   # TODO(slebedev): support tags after we release an initial version.
   return (
-      'https://github.com/google-deepmind/chex/tree/main/chex/%s#L%d#L%d'
+      'https://github.com/google-deepmind/chex/tree/master/chex/%s#L%d#L%d'
       % (
           os.path.relpath(filename, start=os.path.dirname(chex.__file__)),
           lineno,