diff --git a/tests/kubernetes/build_image.sh b/tests/kubernetes/build_image.sh
index 84c3715e1e4..675a15b12bc 100755
--- a/tests/kubernetes/build_image.sh
+++ b/tests/kubernetes/build_image.sh
@@ -32,7 +32,7 @@ while getopts ":pg" opt; do
 done
 
 # Add -gpu to the tag if the GPU image is being built
-if [[ $gpu ]]; then
+if [[ $gpu == "true" ]]; then
   TAG=$TAG-gpu:latest
 else
   TAG=$TAG:latest