diff --git a/tests/grib_set.sh b/tests/grib_set.sh
index 0489c61c3..26cec498b 100755
--- a/tests/grib_set.sh
+++ b/tests/grib_set.sh
@@ -223,8 +223,7 @@ ${tools_dir}/grib_set -s stepUnits=d $input $outfile > $temp 2>&1
 status=$?
 set -e
 [ $status -ne 0 ]
-grep -q "Invalid argument" $temp
-#grep -q "stepUnits: No such code table entry.*Did you mean" $temp
+grep -q "Invalid unit:.*(Unit not found). Available units are:.*" $temp
 
 # ------------------------
 # Unreadable message