diff --git a/test/repeat_unit.sh b/test/repeat_unit.sh
index 58b78e2..0bc0526 100755
--- a/test/repeat_unit.sh
+++ b/test/repeat_unit.sh
@@ -55,7 +55,10 @@ fi
 cd "unit_tests"
 
 # run all test except single one is specified
-if [ ! -z $2 ]; then 
+if [ ! -z $3 ]; then 
+  numberofruns=$2+$3
+  iteration=$3
+elif [ ! -z $2 ]; then 
   numberofruns=$2
 fi