diff --git a/images/jupyter/dockerfile b/images/jupyter/dockerfile
index b202c28..6b89517 100644
--- a/images/jupyter/dockerfile
+++ b/images/jupyter/dockerfile
@@ -2,6 +2,7 @@ ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
 
 USER root
+COPY --chown=appuser:appuser test_buckets.py /home/appuser/work/test_buckets.py
 COPY --chown=appuser:appuser requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir -r /tmp/requirements.txt \
     && rm /tmp/requirements.txt
diff --git a/tests/buckets.py b/images/jupyter/test_buckets.py
similarity index 100%
rename from tests/buckets.py
rename to images/jupyter/test_buckets.py