diff --git a/gm/asyncrescaleandread.cpp b/gm/asyncrescaleandread.cpp
index f61ec15..2e04d27 100644
--- a/gm/asyncrescaleandread.cpp
+++ b/gm/asyncrescaleandread.cpp
@@ -12,7 +12,7 @@
 #include "include/core/SkRect.h"
 #include "include/core/SkSurface.h"
 #include "include/core/SkYUVAIndex.h"
-#include "include/gpu/GrContext.h"
+#include "include/gpu/GrDirectContext.h"
 #include "include/gpu/GrRecordingContext.h"
 #include "src/core/SkAutoPixmapStorage.h"
 #include "src/core/SkConvertPixels.h"
@@ -292,9 +292,12 @@
     SkPaint paint;
     canvas->drawImage(image.get(), 0, 0);
 
+    auto direct = surface->recordingContext() ? surface->recordingContext()->asDirectContext()
+                                              : nullptr;
+
     SkScopeExit scopeExit;
     auto yuvImage = do_read_and_scale_yuv(
-            surface, surface->getContext(), kRec601_SkYUVColorSpace, SkIRect::MakeWH(400, 300),
+            surface, direct, kRec601_SkYUVColorSpace, SkIRect::MakeWH(400, 300),
             {400, 300}, SkImage::RescaleGamma::kSrc, kNone_SkFilterQuality, &scopeExit);
 
     canvas->clear(SK_ColorWHITE);