diff --git a/code_to_optimize/code_directories/simple_tracer_e2e/workload.py b/code_to_optimize/code_directories/simple_tracer_e2e/workload.py
index ef37aab8f..0d74956ce 100644
--- a/code_to_optimize/code_directories/simple_tracer_e2e/workload.py
+++ b/code_to_optimize/code_directories/simple_tracer_e2e/workload.py
@@ -3,13 +3,10 @@
 
 def funcA(number):
     number = min(1000, number)
-    # Use arithmetic formula for sum instead of looping
     k = (number * 100) * (number * 100 - 1) // 2
-    # Simplify the for loop by using sum with a range object (now by formula)
     j = number * (number - 1) // 2
-
-    # Use a map object for efficiency in join (str is faster than formatting and works well here)
-    return " ".join(map(str, range(number)))
+    # List comprehension for slightly better join perf on small N
+    return " ".join([str(i) for i in range(number)])
 
 
 def test_threadpool() -> None: