diff --git a/tests/functional/test_cli_task.py b/tests/functional/test_cli_task.py
index f30d1415ee..bcf26ec22d 100644
--- a/tests/functional/test_cli_task.py
+++ b/tests/functional/test_cli_task.py
@@ -1420,8 +1420,7 @@ class HookTestCase(unittest.TestCase):
         self.assertEqual(
             expected,
             sorted(hook_results,
-                   key=lambda i: list(
-                       i["config"]["trigger"].values())[0]["unit"]))
+                   key=lambda i: i["config"]["trigger"]["args"]["unit"]))
         self._assert_results_time(hook_results)
 
     def test_import_hook_result(self):