diff --git a/Dockerfile b/Dockerfile
deleted file mode 100644
index 540c3ae..0000000
--- a/Dockerfile
+++ /dev/null
@@ -1,22 +0,0 @@
-FROM knthony/run_chrome_driver_in_container
-
-RUN pip install poetry==1.8.3
-
-ENV POETRY_NO_INTERACTION=1 \
-    POETRY_VIRTUALENVS_IN_PROJECT=1 \
-    POETRY_VIRTUALENVS_CREATE=1
-
-RUN mkdir free_llms
-WORKDIR /opt/free_llms
-
-COPY pyproject.toml poetry.lock /opt/free_llms/
-COPY src/free_llms /opt/free_llms/src/free_llms
-COPY tests /opt/free_llms/tests
-RUN poetry install --with dev
-ENV CHROME_VERSION=108
-ENV VIRTUAL_ENV=/opt/free_llms/.venv \
-    PATH="/opt/free_llms/.venv/bin:$PATH"
-
-RUN cp /entrypoint.sh /opt/free_llms
-ENTRYPOINT ["/entrypoint.sh"]
-CMD ["pytest"]
\ No newline at end of file
diff --git a/src/free_llms/models.py b/src/free_llms/models.py
index 3c7a92b..4cea792 100644
--- a/src/free_llms/models.py
+++ b/src/free_llms/models.py
@@ -96,7 +96,7 @@ def check_start_driver(cls, data: Dict) -> Dict:
             if "--window-size" in started_config:
                 raise ValueError("You cannot change the window size in your provided driver config")
         options = configure_options(data["driver_config"] + DRIVERS_DEFAULT_CONFIG)
-        data["driver"] = uc.Chrome(version_main=os.environ.get('CHROME_VERSION',None), options=options, headless=True)
+        data["driver"] = uc.Chrome(options=options, headless=True)
         return data
 
     @property