diff --git a/tests/test_template.py b/tests/test_template.py
index ba0ff2d..35510ad 100644
--- a/tests/test_template.py
+++ b/tests/test_template.py
@@ -25,4 +25,4 @@ class TestTemplate:
         soup = BeautifulSoup(open(output_path), 'html.parser')
         # check onlinetime-data
         assert _format_seconds(clients['1'].onlinetime) == \
-            soup.find('span').text
+            soup.find('span', class_='badge').text