diff --git a/tests/benchmarks/node-infiniband-benchmark.sh b/tests/benchmarks/node-infiniband-benchmark.sh
index 32a93b232..aeb5cdf2b 100755
--- a/tests/benchmarks/node-infiniband-benchmark.sh
+++ b/tests/benchmarks/node-infiniband-benchmark.sh
@@ -46,15 +46,12 @@ INPUT_FILE=$(mktemp)
 LOG_DIR=benchmarks_$(date +%Y%m%d_%H-%M-%S)
 mkdir ${LOG_DIR}
 
-#NUM_VALUES=(3 5 10 25 50)
-#RATE_SAMPLES=(10 100 1000 10000 100000 200000)
-NUM_VALUES=(3)
-RATE_SAMPLES=(100000)
+NUM_VALUES=(3 5 10 25 50)
+RATE_SAMPLES=(10 100 1000 10000 100000 200000)
 TIME_TO_RUN=10
 
 # Declare modes
-#MODES=("TCP" "UDP")
-MODES=("TCP")
+MODES=("TCP" "UDP")
 
 # Initialize counter
 COUNT=0