diff --git a/docker/Dockerfile b/docker/Dockerfile
index 7789e3d9..951f846c 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -3,7 +3,6 @@ FROM python:3.10 as base
 WORKDIR /app
 ADD pyproject.toml .
 ADD requirements.txt .
-RUN pip install --upgrade pip
 RUN pip install . && rm pyproject.toml requirements.txt
 ENV PYTHONPATH=/app
 
diff --git a/requirements.txt b/requirements.txt
index 43380845..ce8145c7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 pinecone-client
-pinecone_datasets
+pinecone_datasets @ git+https://github.com/mrT23/pinecone-datasets.git@main
 dynaconf==3.1.12
 fastapi==0.99.0
 PyGithub==1.59.*