diff --git a/docker/Dockerfile-beta b/docker/Dockerfile-beta
index 781e8544dd..0f1e2a931c 100644
--- a/docker/Dockerfile-beta
+++ b/docker/Dockerfile-beta
@@ -2,7 +2,7 @@ FROM nvcr.io/nvidia/pytorch:24.03-py3
 
 RUN apt update && apt install -y python3.10-venv git-lfs
 
-RUN python3 -m pip install --upgrade pip
+RUN python3 -m pip install --upgrade pip && pip install packaging
 
 RUN groupadd axolotl && useradd -m -g axolotl -s /bin/bash axolotl