content/static: fix unit directories module badge

Change-Id: I2a281ac821778b38df02d232a4ffcb76f76a8e8c
Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/310381
Trust: Jamal Carvalho <jamal@golang.org>
Run-TryBot: Jamal Carvalho <jamal@golang.org>
TryBot-Result: kokoro <noreply+kokoro@google.com>
Reviewed-by: Julie Qiu <julie@golang.org>
diff --git a/content/static/css/unit_directories.css b/content/static/css/unit_directories.css
index 6b7b2b9..3b9c2d1 100644
--- a/content/static/css/unit_directories.css
+++ b/content/static/css/unit_directories.css
@@ -140,3 +140,14 @@
   right: 0;
   text-decoration: none;
 }
+.UnitDirectories-badge {
+  border: 0.0625rem solid var(--gray-4);
+  border-radius: 0.125rem;
+  font-size: 0.6875rem;
+  font-weight: 500;
+  line-height: 1rem;
+  margin-left: 0.5rem;
+  margin-top: 0.125rem;
+  padding: 0 0.35rem;
+  text-align: center;
+}
diff --git a/content/static/html/helpers/_unit_directories.tmpl b/content/static/html/helpers/_unit_directories.tmpl
index 6f16641..a6c2c27 100644
--- a/content/static/html/helpers/_unit_directories.tmpl
+++ b/content/static/html/helpers/_unit_directories.tmpl
@@ -47,7 +47,7 @@
           {{- end -}}
           {{- if .Root -}}
             <a href="{{.Root.URL}}">{{.Root.Suffix}}</a>
-            {{if .Root.IsModule}}<span class="UnitHeader-badge">module</span>{{end}}
+            {{if .Root.IsModule}}<span class="UnitDirectories-badge">module</span>{{end}}
           </div>
           <div class="UnitDirectories-mobileSynopsis">{{.Root.Synopsis}}</div>
         </div>
@@ -65,7 +65,7 @@
         <div class="UnitDirectories-subdirectory">
           <span>
             <a href="{{.URL}}">{{.Suffix}}</a>
-            {{if .IsModule}}<span class="UnitHeader-badge">Module</span>{{end}}
+            {{if .IsModule}}<span class="UnitDirectories-badge">Module</span>{{end}}
           </span>
           <div class="UnitDirectories-mobileSynopsis">{{.Synopsis}}</div>
         </div>