diff --git a/dm/DMGpuTestProcs.cpp b/dm/DMGpuTestProcs.cpp
index aba40a4..e789a33 100644
--- a/dm/DMGpuTestProcs.cpp
+++ b/dm/DMGpuTestProcs.cpp
@@ -130,13 +130,12 @@
         }
 
         skiatest::graphite::ContextInfo ctxInfo = factory.getContextInfo(contextType);
-        skgpu::graphite::Context* context = ctxInfo.fContext;
-        if (!context) {
+        if (!ctxInfo.fContext) {
             continue;
         }
 
         ReporterContext ctx(reporter, SkString(skgpu::ContextTypeName(contextType)));
-        (*test)(reporter, context);
+        (*test)(reporter, ctxInfo.fContext, ctxInfo.fTestContext);
     }
 }