diff --git a/gapis/api/vulkan/replay.go b/gapis/api/vulkan/replay.go
index 19bb91df2..4ca06b685 100644
--- a/gapis/api/vulkan/replay.go
+++ b/gapis/api/vulkan/replay.go
@@ -1050,7 +1050,7 @@ func (a API) Replay(
 			makeReadable.imagesOnly = true
 			optimize = false
 			transforms.Add(NewWaitForPerfetto(req.traceOptions, req.handler, req.buffer))
-			if strings.Contains(device.GetConfiguration().GetHardware().GetGPU().GetName(), "Adreno") {
+			if device.GetConfiguration().GetPerfettoCapability().GetGpuProfiling().GetHasRenderStageProducerLayer() {
 				transforms.Add(&profilingLayers{})
 			}
 			transforms.Add(replay.NewMappingExporter(ctx, req.handleMappings))