diff --git a/Dockerfile b/Dockerfile
index 01f8d68..5392ac3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -79,6 +79,9 @@ COPY --from=ghcr.io/ublue-os/akmods-nvidia:38-535 /rpms /tmp/akmods-rpms
 RUN rpm-ostree install \
     /tmp/akmods-rpms/ublue-os/ublue-os-nvidia-addons-*.rpm
 
+# install kernel headers
+RUN rpm-ostree install kernel-cachyos-lts-headers
+
 RUN rpm-ostree install \
     xorg-x11-drv-nvidia{,-cuda,-devel,-kmodsrc} \
     xorg-x11-drv-nvidia-libs.i686 \
@@ -99,9 +102,7 @@ COPY install-nvidia.sh /tmp/install-nvidia.sh
 # Enable cliwrap.
 RUN rpm-ostree cliwrap install-to-root / && \
 # Replace the kernel, kernel-core and kernel-modules packages.
-rpm-ostree override remove kernel kernel-core kernel-modules kernel-modules-core kernel-modules-extra --install kernel-cachyos-lts && \
-# install kernel headers
-rpm-ostree install kernel-cachyos-lts-headers
+rpm-ostree override remove kernel kernel-core kernel-modules kernel-modules-core kernel-modules-extra --install kernel-cachyos-lts
 
 # install akmods
 RUN ls /tmp/nvidia && /tmp/install-nvidia.sh