diff --git a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/amd.sh b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/amd.sh
index 42e5ca7..365e8ab 100755
--- a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/amd.sh
+++ b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/amd.sh
@@ -13,9 +13,9 @@ python_lock=$(micromamba -n $MAMBA_DEFAULT_ENV run python -V|awk '{print $2}'|cu
 $MAMBA_INSTALL -n $MAMBA_DEFAULT_ENV \
     -c pytorch \
     -c conda-forge \
-    pytorch=${PYTORCH_VERSION} torchvision torchaudio \
-    python=${python_lock} \
-    --only-deps
+        pytorch=${PYTORCH_VERSION} torchvision torchaudio \
+        python=${python_lock} \
+        --only-deps
 # Now pip install...
 micromamba -n $MAMBA_DEFAULT_ENV run pip install \
     --no-cache-dir \
diff --git a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh
index 9b9d402..4280043 100755
--- a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh
+++ b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/common.sh
@@ -4,5 +4,9 @@
 set -eo pipefail
 
 # For logic common to cpu, cuda & rocm
+$MAMBA_INSTALL -n $MAMBA_DEFAULT_ENV \
+    -c pytorch \
+    -c conda-forge \
+        ffmpeg">=4.3.1"
 
 exit 0
\ No newline at end of file
diff --git a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/cpu.sh b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/cpu.sh
index 407e4b4..1a02dc9 100755
--- a/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/cpu.sh
+++ b/build/COPY_ROOT/opt/ai-dock/bin/build/layer0/cpu.sh
@@ -10,6 +10,8 @@ python_lock=$(micromamba -n $MAMBA_DEFAULT_ENV run python -V|awk '{print $2}'|cu
 $MAMBA_INSTALL -n $MAMBA_DEFAULT_ENV \
     -c pytorch \
     -c conda-forge \
-    python=${python_lock} \
-    pytorch=${PYTORCH_VERSION} torchvision torchaudio cpuonly \
+        python=${python_lock} \
+        pytorch=${PYTORCH_VERSION} torchvision \
+        torchaudio \
+        cpuonly
     
\ No newline at end of file