diff --git a/docs/conf.py b/docs/conf.py
index 03c0cb7436..4f1f8ea971 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -116,7 +116,7 @@
   href="https://flax-nnx.readthedocs.io/en/latest/index.html"
   style="text-decoration: none; color: white;"
 >
-  This is Flax Linen. Explore the new <b>Flax NNX</b> API ✨
+  <span style="color: lightgray;">[Flax Linen]</span> Explore the new <b>Flax NNX</b> API ✨
 </a>
 """
 
diff --git a/docs_nnx/conf.py b/docs_nnx/conf.py
index 725b6890d0..8d541f6110 100644
--- a/docs_nnx/conf.py
+++ b/docs_nnx/conf.py
@@ -116,7 +116,7 @@
   href="https://flax-linen.readthedocs.io/en/latest"
   style="text-decoration: none; color: white;"
 >
-  This Flax NNX. Click here for <b>Flax Linen</b>.
+  <span style="color: lightgray;">[Flax NNX]</span> Click here for the old <b>Flax Linen</b>
 </a>
 """