diff --git a/hw/xbox/nv2a/pgraph/vk/surface-compute.c b/hw/xbox/nv2a/pgraph/vk/surface-compute.c
index e59a44f07d..ebbaca65a8 100644
--- a/hw/xbox/nv2a/pgraph/vk/surface-compute.c
+++ b/hw/xbox/nv2a/pgraph/vk/surface-compute.c
@@ -139,7 +139,7 @@ static gchar *get_compute_shader_glsl(VkFormat host_fmt, bool pack,
 
     gchar *glsl = g_strdup_printf(
         "#version 450\n"
-        "layout(local_size_x = %d) in;\n"
+        "layout(local_size_x = %d, local_size_y = 1, local_size_z = 1) in;\n"
         "%s", workgroup_size, template);
     assert(glsl);