diff --git a/tools/pipeline_perf_test/test_suites/integration/continuous/100klrps-docker.yaml b/tools/pipeline_perf_test/test_suites/integration/continuous/100klrps-docker.yaml
index 911fa79dee..5381c1a6c7 100644
--- a/tools/pipeline_perf_test/test_suites/integration/continuous/100klrps-docker.yaml
+++ b/tools/pipeline_perf_test/test_suites/integration/continuous/100klrps-docker.yaml
@@ -82,6 +82,9 @@ tests:
         loadgen_exporter_type: otlp
         backend_receiver_type: otlp
         max_batch_size: 1000
+        data_source: static
+        generation_strategy: fresh
+        use_trace_context: true
   - name: OTLP-ATTR-OTAP
     from_template:
       path: test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
@@ -90,6 +93,9 @@ tests:
         loadgen_exporter_type: otlp
         backend_receiver_type: otap
         max_batch_size: 1000
+        data_source: static
+        generation_strategy: fresh
+        use_trace_context: true
   - name: OTAP-ATTR-OTAP
     from_template:
       path: test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
@@ -98,6 +104,9 @@ tests:
         loadgen_exporter_type: otap
         backend_receiver_type: otap
         max_batch_size: 1000
+        data_source: static
+        generation_strategy: fresh
+        use_trace_context: true
   - name: OTAP-ATTR-OTLP
     from_template:
       path: test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
@@ -106,6 +115,9 @@ tests:
         loadgen_exporter_type: otap
         backend_receiver_type: otlp
         max_batch_size: 1000
+        data_source: static
+        generation_strategy: fresh
+        use_trace_context: true
   - name: OTLP-BATCH-OTLP
     from_template:
       path: test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
@@ -114,6 +126,9 @@ tests:
         loadgen_exporter_type: otlp
         backend_receiver_type: otlp
         max_batch_size: 1000
+        data_source: static
+        generation_strategy: fresh
+        use_trace_context: true
         exporter_extra_config:
           max_in_flight: 256
   - name: OTAP-BATCH-OTAP
@@ -124,3 +139,6 @@ tests:
         loadgen_exporter_type: otap
         backend_receiver_type: otap
         max_batch_size: 1000
+        data_source: static
+        generation_strategy: fresh
+        use_trace_context: true
diff --git a/tools/pipeline_perf_test/test_suites/integration/templates/configs/loadgen/config.yaml.j2 b/tools/pipeline_perf_test/test_suites/integration/templates/configs/loadgen/config.yaml.j2
index fccee5ce90..ee0f0b16c6 100644
--- a/tools/pipeline_perf_test/test_suites/integration/templates/configs/loadgen/config.yaml.j2
+++ b/tools/pipeline_perf_test/test_suites/integration/templates/configs/loadgen/config.yaml.j2
@@ -16,7 +16,7 @@ groups:
           receiver:
             type: "urn:otel:receiver:traffic_generator"
             config:
-              data_source: semantic_conventions
+              data_source: {{ data_source | default("semantic_conventions") }}
               generation_strategy: {{ generation_strategy | default("pre_generated") }}
               traffic_config:
                 max_batch_size: {{ max_batch_size | default(1000) }}
@@ -24,6 +24,9 @@ groups:
                 metric_weight: {{ metric_weight | default(0) }}
                 trace_weight: {{ trace_weight | default(0) }}
                 log_weight: {{ log_weight | default(100) }}
+                {%- if use_trace_context is defined and use_trace_context %}
+                use_trace_context: true
+                {%- endif %}
 
 {%- set exporter_type = loadgen_exporter_type | default("otap") %}
 {%- set engine_host = engine_hostname | default("localhost") %}
diff --git a/tools/pipeline_perf_test/test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml b/tools/pipeline_perf_test/test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
index ef1bc3722b..7604ce4ec9 100644
--- a/tools/pipeline_perf_test/test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
+++ b/tools/pipeline_perf_test/test_suites/integration/templates/test_steps/df-loadgen-steps-docker.yaml
@@ -67,7 +67,9 @@ steps:
                 # ----- Receiver traffic config -----
                 max_batch_size: {{max_batch_size | default(1000)}}
                 signals_per_second: {% if signals_per_second is none %}null{% else %}{{signals_per_second | default(100000)}}{% endif %}
+                data_source: {{data_source | default("semantic_conventions")}}
                 generation_strategy: {{generation_strategy | default("pre_generated")}}
+                {% if use_trace_context is defined %}use_trace_context: {{use_trace_context}}{% endif %}
                 metric_weight: {{metric_weight | default(0)}}
                 trace_weight: {{trace_weight | default(0)}}
                 log_weight: {{log_weight | default(100)}}