diff --git a/Dockerfile.ubi b/Dockerfile.ubi
index ae6bd0bc6dfd2..6676f9da6be28 100644
--- a/Dockerfile.ubi
+++ b/Dockerfile.ubi
@@ -193,7 +193,7 @@ FROM vllm-openai as vllm-grpc-adapter
 USER root
 
 RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install vllm-tgis-adapter==0.2.3
+    pip install git+https://github.com/maxdebayser/vllm-tgis-adapter@default_add_special_tokens
 
 ENV GRPC_PORT=8033
 USER 2000