diff --git a/tests/integration/test_flatten_keywords.py b/tests/integration/test_flatten_keywords.py
index 199e73f..4d6a021 100644
--- a/tests/integration/test_flatten_keywords.py
+++ b/tests/integration/test_flatten_keywords.py
@@ -33,24 +33,24 @@
             0,
             "Content flattened.",
         ),
-        # (
-        #     "examples/for_keyword.robot",
-        #     "FOR",
-        #     0,
-        #     ["PASSED"] * 5,
-        #     2,
-        #     0,
-        #     "2 passing items flattened using the --flatten-keywords option.",
-        # ),
-        # (
-        #     "examples/while_keyword.robot",
-        #     "WHILE",
-        #     0,
-        #     ["PASSED"] * 7,
-        #     5,
-        #     2,
-        #     "2 passing items flattened using the --flatten-keywords option.",
-        # ),
+        (
+            "examples/for_keyword.robot",
+            "FOR",
+            0,
+            ["PASSED"] * 3,
+            4,
+            0,
+            "Content flattened.",
+        ),
+        (
+            "examples/while_keyword.robot",
+            "WHILE",
+            0,
+            ["PASSED"] * 4,
+            9,
+            2,
+            "Content flattened.",
+        ),
         # (
         #     "examples/for_keyword_failed.robot",
         #     "FOR",