diff --git a/docker/setup.sh b/docker/setup.sh
index c839a0f..2f6116a 100755
--- a/docker/setup.sh
+++ b/docker/setup.sh
@@ -9,7 +9,7 @@ git clone https://github.com/Pelochus/ezrknn-llm/
 curl https://raw.githubusercontent.com/Pelochus/ezrknn-llm/master/install.sh | sudo bash
 
 # For running the test.py
-pip install /ezrknn-llm/rkllm-toolkit/packages/rkllm_toolkit-1.0.0-cp38-cp38-linux_x86_64.whl
+pip install /ezrknn-llm/rkllm-toolkit/packages/rkllm_toolkit-1.0.1-cp38-cp38-linux_x86_64.whl
 
 # Clone some compatible LLMs
 cd /ezrknn-llm/rkllm-toolkit/examples/huggingface