diff --git a/codeflash/verification/test_results.py b/codeflash/verification/test_results.py
index c7a210a6a..874c38072 100644
--- a/codeflash/verification/test_results.py
+++ b/codeflash/verification/test_results.py
@@ -33,7 +33,7 @@ class TestType(Enum):
     INIT_STATE_TEST = 6
 
     def to_name(self) -> str:
-        if self == TestType.INIT_STATE_TEST:
+        if self is TestType.INIT_STATE_TEST:
             return ""
         names = {
             TestType.EXISTING_UNIT_TEST: "⚙️ Existing Unit Tests",
@@ -166,6 +166,8 @@ def report_to_string(report: dict[TestType, dict[str, int]]) -> str:
     def report_to_tree(report: dict[TestType, dict[str, int]], title: str) -> Tree:
         tree = Tree(title)
         for test_type in TestType:
+            if test_type is TestType.INIT_STATE_TEST:
+                continue
             tree.add(
                 f"{test_type.to_name()} - Passed: {report[test_type]['passed']}, Failed: {report[test_type]['failed']}"
             )