diff --git a/data/tools/unit_tree/html_output.py b/data/tools/unit_tree/html_output.py
index 79665acc689..d150e3d28ab 100644
--- a/data/tools/unit_tree/html_output.py
+++ b/data/tools/unit_tree/html_output.py
@@ -11,7 +11,7 @@ pics_location = "../../pics"
html_header = '''
-
+
@@ -23,7 +23,7 @@ html_header = '''
- %(title)s
+ %(title)s - Wesnoth Units Database
@@ -730,7 +730,7 @@ class HTMLOutput:
def write_units_tree(self, grouper, title, add_parents):
self.output.write(html_header % {"path": "../../",
- "title": title})
+ "title": title, "classes": "wmlunits-tree"})
n = self.analyze_units(grouper, add_parents)
self.write_navbar("units_tree")
@@ -781,7 +781,7 @@ class HTMLOutput:
self.output = output
write(html_header % {"path": "../../",
- "title": display_name})
+ "title": display_name, "classes": "wmlunits-unit"})
self.write_navbar("unit_report")
self.output.write("")
diff --git a/data/tools/unit_tree/overview.py b/data/tools/unit_tree/overview.py
index 50191016e86..8ae6af18022 100755
--- a/data/tools/unit_tree/overview.py
+++ b/data/tools/unit_tree/overview.py
@@ -54,8 +54,9 @@ def main(folder):
def w(x): out.write(x + "\n")
path = ""
- title = "Wesnoth Unit Database Overview"
+ title = "Database Build Report"
generation_note = "generated on " + time.ctime()
+ classes = "wmlunits-report"
w(html_output.html_header % locals())