diff --git a/data/tools/unit_tree/animations.py b/data/tools/unit_tree/animations.py
index 9eabb3fccdf..4c8bf8e4bb4 100644
--- a/data/tools/unit_tree/animations.py
+++ b/data/tools/unit_tree/animations.py
@@ -156,11 +156,11 @@ td.none {border: solid 1px; background-color: #ffffff;}
f.write("total frames (number of animations)\n")
f.write("
Mainline
\n")
- us = [x for x in wesnoth.unit_lookup.values() if x.campaign == "mainline"]
+ us = [x for x in wesnoth.unit_lookup.values() if x.campaigns[0] == "mainline"]
put_units(f, us)
f.write("Campaigns and Addons
\n")
- us = [x for x in wesnoth.unit_lookup.values() if x.campaign != "mainline"]
+ us = [x for x in wesnoth.unit_lookup.values() if x.campaigns[0] != "mainline"]
put_units(f, us)
f.write("