diff --git a/src/nouveau/compiler/nak.rs b/src/nouveau/compiler/nak.rs
index 4e2224db9e5..e04337585e3 100644
--- a/src/nouveau/compiler/nak.rs
+++ b/src/nouveau/compiler/nak.rs
@@ -496,7 +496,7 @@ pub extern "C" fn nak_compile_shader(
             ],
             smem_size: nir.info.shared_size.try_into().unwrap(),
         },
-        hdr: encode_hdr_for_nir(nir, nir.scratch_size),
+        hdr: encode_hdr_for_nir(nir, s.tls_size, fs_key),
     };
 
     let code = if nak.sm >= 75 {