diff --git a/run-ollama-docker.sh b/run-ollama-docker.sh
index c8ce166..c2a025b 100644
--- a/run-ollama-docker.sh
+++ b/run-ollama-docker.sh
@@ -10,10 +10,10 @@ docker pull ollama/ollama:latest
 
 docker_args="-d -v ollama:/root/.ollama -p $host_port:$container_port --name ollama ollama/ollama"
 
-if [ "$use_gpu" == "y" ]; then
-    docker_args+=" --gpus=all"
+if [ "$use_gpu" = "y" ]; then
+    docker_args="--gpus=all $docker_args"
 fi
 
-docker run "$docker_args"
+docker run $docker_args
 
-docker image prune -f
+docker image prune -f
\ No newline at end of file