diff --git a/src/dxvk/shaders/rtx/pass/integrate/integrate_nee.comp.slang b/src/dxvk/shaders/rtx/pass/integrate/integrate_nee.comp.slang
index 9cd74b367..a214f7cf3 100644
--- a/src/dxvk/shaders/rtx/pass/integrate/integrate_nee.comp.slang
+++ b/src/dxvk/shaders/rtx/pass/integrate/integrate_nee.comp.slang
@@ -162,7 +162,7 @@ void main(uint2 threadIndex : SV_DispatchThreadID, uint2 LocalIndex : SV_GroupTh
           if (any(diffuseLight + specularLight) > 0 && (threadIndex.x + threadIndex.y) % 4 == 0)
           {
             float16_t accumulateValue = calcBt709Luminance(diffuseLight + specularLight);
-            int prefixTask = NEECacheUtils.convertIDToPrefixSumID(surfaceIndex, primitiveIndex, PrimitiveIDPrefixSum);
+            int prefixTask = NEECacheUtils.convertIDToPrefixSumID(candidate.getSurfaceID(), primitiveIndex, PrimitiveIDPrefixSum);
             float16_t randomOffset = RAB_GetNextRandom(rtxdiRNG);
             cell.insertSlotTask(prefixTask, accumulateValue, randomOffset, false);
           }