diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 7893c1e..b18718b 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -9,19 +9,21 @@ RUN apt update && apt upgrade -y && apt install -y \
     curl \
     wget \
     golang-go \
+    ca-certificates
+
+# Directly install glslc via Debian arm64 package
+RUN wget http://ftp.debian.org/debian/pool/main/s/shaderc/glslc_2023.2-1_arm64.deb \
+    && apt-get update && apt-get install -y ./glslc_2023.2-1_arm64.deb \
+    && rm glslc_2023.2-1_arm64.deb
+
+# install the rest of the requirements
+RUN apt update && apt upgrade -y && apt install -y \
     libvulkan-dev \
     vulkan-tools \
     glslang-tools \
     ninja-build \
     spirv-headers \
-    spirv-tools \
-    ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
-
-# Directly install glslc via Debian arm64 package
-RUN wget http://ftp.debian.org/debian/pool/main/s/shaderc/glslc_2023.2-1_arm64.deb \
-    && apt-get update && apt-get install -y ./glslc_2023.2-1_arm64.deb \
-    && rm glslc_2023.2-1_arm64.deb && rm -rf /var/lib/apt/lists/*
+    spirv-tools
 
 WORKDIR /app
 RUN git clone --recursive https://github.com/ollama/ollama.git .