diff --git a/pkg/cmd/drtprod/configs/drt_scale.yaml b/pkg/cmd/drtprod/configs/drt_scale.yaml
index 86a146c24d76..adc644ee31c4 100644
--- a/pkg/cmd/drtprod/configs/drt_scale.yaml
+++ b/pkg/cmd/drtprod/configs/drt_scale.yaml
@@ -172,5 +172,7 @@ targets:
         flags:
           scale-factor: 1000
       - script: "pkg/cmd/drtprod/scripts/generate_tpch_run.sh"
+        args:
+          - scale_factor_1000
         flags:
           scale-factor: 1000
diff --git a/pkg/cmd/drtprod/scripts/generate_tpch_run.sh b/pkg/cmd/drtprod/scripts/generate_tpch_run.sh
index 4cc3befdb509..1a13a0e52c21 100755
--- a/pkg/cmd/drtprod/scripts/generate_tpch_run.sh
+++ b/pkg/cmd/drtprod/scripts/generate_tpch_run.sh
@@ -8,7 +8,7 @@
 # This script sets up the tpch run workload script in the workload nodes
 # The --scale-factor and other flags are passed as argument to this script
 # NOTE - This uses CLUSTER and WORKLOAD_CLUSTER environment variable, if not set the script fails
-if [ "$#" -lt 3 ]; then
+if [ "$#" -lt 2 ]; then
   echo "Usage: $0 <script_suffix> <flags to init:--scale-factor,--db>"
   exit 1
 fi