diff --git a/tests/test_sub_class_name.py b/tests/test_sub_class_name.py
index d067002..b12f161 100644
--- a/tests/test_sub_class_name.py
+++ b/tests/test_sub_class_name.py
@@ -62,7 +62,7 @@ def _find_failing_node_model(module_node):
     "viewset_model": {
         "versions": ">=2",
         "msg_id": "nb-sub-class-name",
-        "line": 5,
+        "line": 6,
         "col_offset": 0,
         "args": ("AddressObjectUIViewSet",),
         "node": _find_failing_node_model,
@@ -78,7 +78,7 @@ def _find_failing_node_model(module_node):
     "table_model": {
         "versions": ">=2",
         "msg_id": "nb-sub-class-name",
-        "line": 5,
+        "line": 6,
         "col_offset": 0,
         "args": ("AddressObjectTable",),
         "node": _find_failing_node_model,