diff --git a/ci/run_cpp_test.sh b/ci/run_cpp_test.sh
index 2b4714d6a6af1f17425f894ddafea9040ac24692..1e0e16ebc948b21f90dc6752c5425c3905545af7 100755
--- a/ci/run_cpp_test.sh
+++ b/ci/run_cpp_test.sh
@@ -3,7 +3,6 @@
 set -e
 
 BASEDIR=$(readlink -f "$(dirname "$0")"/..)
-source "${BASEDIR}/ci/utils.sh"
 
 export MGB_TEST_NO_LOG=1
 export MGB_STABLE_RNG=1
@@ -16,9 +15,9 @@ function cpp_test {
 }
 
 
-if [[ "$1" != "cpu" && "$1" != "cuda" ]] ; then
+if [[ "$1" == "cpu" || "$1" == "cuda" ]] ; then
+    cpp_test "$@"
+else
     echo "Argument must cpu or cuda"
     exit 1
-fi
-
-cpp_test "$@"
\ No newline at end of file
+fi
\ No newline at end of file