diff --git a/Runtime/FrameInterpolation/FrameInterpolationContext.cs b/Runtime/FrameInterpolation/FrameInterpolationContext.cs
index db84dfd..fdb751f 100644
--- a/Runtime/FrameInterpolation/FrameInterpolationContext.cs
+++ b/Runtime/FrameInterpolation/FrameInterpolationContext.cs
@@ -282,7 +282,7 @@ namespace FidelityFX.FrameGen
             const int resolutionMultiplier = 1;
             
             RectInt rectInfo = new RectInt(0, 0, resolution.x * resolutionMultiplier, resolution.y * resolutionMultiplier);
-            SpdSetup(rectInfo, out dispatchThreadGroupCount, out var workGroupOffset, out var numWorkGroupsAndMips, 4);
+            SpdSetup(rectInfo, out dispatchThreadGroupCount, out var workGroupOffset, out var numWorkGroupsAndMips);
             
             ref FrameInterpolation.InpaintingPyramidConstants spdConstants = ref SpdConstants;
             spdConstants.numWorkGroups = (uint)numWorkGroupsAndMips.x;