diff --git a/tests/integration/test_before_after.py b/tests/integration/test_before_after.py
index af6622c..142aa12 100644
--- a/tests/integration/test_before_after.py
+++ b/tests/integration/test_before_after.py
@@ -23,8 +23,13 @@
 @pytest.mark.parametrize(
     "test, idx_to_check, step_name, suite_name",
     [
-        ("examples/before_after/before_suite_with_steps.robot", 1, "Log suite setup", "Before Suite With Steps"),
-        ("examples/before_after/after_suite_with_steps.robot", 3, "Log suite tear down", "After Suite With Steps"),
+        ("examples/before_after/before_suite_with_steps.robot", 1, "SETUP Log suite setup", "Before Suite With Steps"),
+        (
+            "examples/before_after/after_suite_with_steps.robot",
+            3,
+            "TEARDOWN Log suite tear down",
+            "After Suite With Steps",
+        ),
     ],
 )
 @mock.patch(REPORT_PORTAL_SERVICE)