diff --git a/sail-partitioned/benchmark.sh b/sail-partitioned/benchmark.sh
index c65266e46..dec7af122 100755
--- a/sail-partitioned/benchmark.sh
+++ b/sail-partitioned/benchmark.sh
@@ -45,7 +45,7 @@ echo "Install Python packages"
 python3 -m venv myenv
 source myenv/bin/activate
 pip install --upgrade setuptools wheel
-pip install --no-cache-dir "pysail==0.5.2"
+pip install --no-cache-dir "pysail>=0.6.0,<0.7.0"
 pip install "pyspark-client==4.1.1" \
   pandas \
   psutil
diff --git a/sail/benchmark.sh b/sail/benchmark.sh
index 3e81894a6..982465257 100755
--- a/sail/benchmark.sh
+++ b/sail/benchmark.sh
@@ -45,7 +45,7 @@ echo "Install Python packages"
 python3 -m venv myenv
 source myenv/bin/activate
 pip install --upgrade setuptools wheel
-pip install --no-cache-dir "pysail==0.5.2"
+pip install --no-cache-dir "pysail>=0.6.0,<0.7.0"
 pip install "pyspark-client==4.1.1" \
   pandas \
   psutil