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 1c6a3f1f4..21db38678 100644
--- a/code_to_optimize/code_directories/simple_tracer_e2e/workload.py
+++ b/code_to_optimize/code_directories/simple_tracer_e2e/workload.py
@@ -42,7 +42,8 @@ def _extract_features(self, x):
 
     def _classify(self, features):
         total = sum(features)
-        return [total % self.num_classes for _ in features]
+        mod_val = total % self.num_classes
+        return [mod_val] * len(features)
 
 
 class SimpleModel: