diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index 8f541e0..72abc1b 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -55,5 +55,6 @@ jobs:
     - name: Build and Push to Docker Hub
       uses: docker/build-push-action@v2.6.1
       with:
+        context: .
         push: true
         tags: qmkfm/qmk_cli:latest
diff --git a/Dockerfile b/Dockerfile
index 5b5a7e1..ceca25a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,8 +1,13 @@
 FROM qmkfm/base_container
 
+# Copy package in
+COPY dist /tmp
+
 # Install python packages
-RUN python3 -m pip install --upgrade pip setuptools wheel
-RUN python3 -m pip install --upgrade nose2 qmk
+RUN python3 -m pip uninstall qmk || true
+RUN python3 -m pip install --upgrade pip setuptools wheel nose2 && \
+    python3 -m pip install /tmp/dist/qmk-*.whl && \
+    rm -rf /tmp/dist
 
 # Set the default location for qmk_firmware
 ENV QMK_HOME /qmk_firmware