diff --git a/scripts/setup.sh b/scripts/setup.sh
index 2b3b8a28..68b98779 100755
--- a/scripts/setup.sh
+++ b/scripts/setup.sh
@@ -200,7 +200,7 @@ install_graphrag() {
     
     # Check the current graphrag version in Poetry environment
     log_step "Checking graphrag version in Poetry environment"
-    GRAPHRAG_VERSION=$(poetry run pip show graphrag 2>/dev/null | grep "Version:" | cut -d " " -f2)
+    GRAPHRAG_VERSION=$(poetry run pip show graphrag 2>/dev/null | awk '/^Versi  on: / {print $2; exit}')
     GRAPHRAG_TARGET="1.2.1.dev27"
     GRAPHRAG_LOCAL_PATH="dependencies/graphrag-${GRAPHRAG_TARGET}.tar.gz"