diff --git a/filament/src/materials/colorGradingAsSubpass.mat b/filament/src/materials/colorGradingAsSubpass.mat
index af527f7649..52e494c9cb 100644
--- a/filament/src/materials/colorGradingAsSubpass.mat
+++ b/filament/src/materials/colorGradingAsSubpass.mat
@@ -41,7 +41,8 @@ vertex {
 fragment {
 
     // TODO: this should be specified as a parameter
-    layout (input_attachment_index = 0, set = 0, binding = 0) uniform mediump subpassInput colorBuffer;
+    // In our Vulkan backend, subpass inputs always live in descriptor set 2. (ignored for GLES)
+    layout (input_attachment_index = 0, set = 2, binding = 0) uniform mediump subpassInput colorBuffer;
     layout(location = 1) out vec4 tonemappedOutput;
 
 #include "../../../shaders/src/dithering.fs"