diff --git a/projects/rocprofiler-sdk/tests/counter-collection/validate.py b/projects/rocprofiler-sdk/tests/counter-collection/validate.py
index 93fa53d3b9..ceded9efec 100644
--- a/projects/rocprofiler-sdk/tests/counter-collection/validate.py
+++ b/projects/rocprofiler-sdk/tests/counter-collection/validate.py
@@ -26,16 +26,13 @@ def test_counter_values(input_data):
 
     scaling_factor = 1
     for itr in data["rocprofiler-sdk-json-tool"]["agents"]:
-        if itr["type"] != 2 or itr["wave_front_size"] < 0:
-            continue
+        if itr["type"] == 2 and itr["wave_front_size"] > 0:
+            scaling_factor = 64 / itr["wave_front_size"]
+            break
 
-        scaling_factor = 64 / itr["wave_front_size"]
-
-        for itr in data["rocprofiler-sdk-json-tool"]["buffer_records"]["counter_collection"]:
-            value = itr["counter_value"]
-            assert int(round(value, 0)) == int(round(1 * scaling_factor, 0)), (f"Failure on agent "
-                f"{str(itr)} expected {1 * scaling_factor} but got {value} "
-                f"Debug Info:\n {str(data)}")
+    for itr in data["rocprofiler-sdk-json-tool"]["buffer_records"]["counter_collection"]:
+        value = itr["counter_value"]
+        assert int(round(value, 0)) == int(round(1 * scaling_factor, 0)), str(data)
 
 
 if __name__ == "__main__":