diff --git a/modules/jupyter/authentication/Dockerfile b/modules/jupyter/authentication/Dockerfile
new file mode 100644
index 000000000..d86ad5992
--- /dev/null
+++ b/modules/jupyter/authentication/Dockerfile
@@ -0,0 +1,9 @@
+FROM jupyterhub/k8s-hub:3.3.8
+
+ADD ./authenticator/ ./authenticator
+
+USER root
+RUN chmod -R u+w ./authenticator
+
+RUN pip3 install notebook
+RUN pip3 install ./authenticator
\ No newline at end of file
diff --git a/modules/jupyter/authentication/authenticator/setup.py b/modules/jupyter/authentication/authenticator/setup.py
index 3c67b68d5..3f2e5278e 100644
--- a/modules/jupyter/authentication/authenticator/setup.py
+++ b/modules/jupyter/authentication/authenticator/setup.py
@@ -23,10 +23,10 @@
     long_description="long description here",
     long_description_content_type="text/markdown",
     url="https://github.com/chiayi/ai-on-gke/tree/main/ray-on-gke/user/jupyterhub/auth",
-    packages=['gcpiapjwtauthenticator'],
+    packages=setuptools.find_packages(),
     license='Apache 2.0',
     install_requires=[
-        "jupyterhub>=4.1.0",
+        "jupyterhub",
         "tornado>=6.3.3",
         'oauthenticator>=0.9.0',
         'pyjwt>=2.7.0',
diff --git a/modules/jupyter/authentication/docker_image/cloudbuild.yaml b/modules/jupyter/authentication/cloudbuild.yaml
similarity index 100%
rename from modules/jupyter/authentication/docker_image/cloudbuild.yaml
rename to modules/jupyter/authentication/cloudbuild.yaml
diff --git a/modules/jupyter/authentication/docker_image/Dockerfile b/modules/jupyter/authentication/docker_image/Dockerfile
deleted file mode 100644
index fbcdc4411..000000000
--- a/modules/jupyter/authentication/docker_image/Dockerfile
+++ /dev/null
@@ -1,3 +0,0 @@
-FROM jupyterhub/k8s-hub:3.3.0
-
-RUN pip3 install --no-cache-dir git+https://github.com/GoogleCloudPlatform/ai-on-gke/@main#subdirectory=modules/jupyter/authentication/authenticator
\ No newline at end of file