diff --git a/src/start.sh b/src/start.sh
index c557a56..b0e7d95 100644
--- a/src/start.sh
+++ b/src/start.sh
@@ -30,7 +30,7 @@ while ! check_server_is_running; do
     sleep 5
 done
 # IF $MODEL_NAME is set, make sure to pull the model, else just skip
-if [ -z "$MODEL_NAME" ]; then
+if [ -z "$OLLAMA_MODEL_NAME" ]; then
     echo "No model name provided. Skipping model pull..."
 else
     echo "Pulling model $MODEL_NAME..."