diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 2d9969524..b14d080ba 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -48,6 +48,7 @@ jobs:
 
       - name: Install dependencies
         run: |
+          pip3 install packaging
           pip3 install -U -e .
           pip3 install -r requirements-tests.txt
 
diff --git a/docker/Dockerfile b/docker/Dockerfile
index efc40ab06..f8b97771d 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -18,6 +18,8 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
 
 WORKDIR /workspace/axolotl
 
+RUN pip install packaging
+
 # If AXOLOTL_EXTRAS is set, append it in brackets
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
         pip install -e .[deepspeed,flash-attn,mamba-ssm,$AXOLOTL_EXTRAS]; \