diff --git a/tests/WrappedSurfaceCopyOnWriteTest.cpp b/tests/WrappedSurfaceCopyOnWriteTest.cpp
index 1a91568..6ec2988 100644
--- a/tests/WrappedSurfaceCopyOnWriteTest.cpp
+++ b/tests/WrappedSurfaceCopyOnWriteTest.cpp
@@ -36,8 +36,8 @@
     };
 
     auto surfaceProxyID = [&](const sk_sp<SkSurface>& surf) {
-        GrSurfaceDrawContext* sdc = SkCanvasPriv::TopDeviceSurfaceDrawContext(surf->getCanvas());
-        return sdc->asSurfaceProxy()->uniqueID();
+        GrRenderTargetProxy* rtp = SkCanvasPriv::TopDeviceTargetProxy(surf->getCanvas());
+        return rtp->uniqueID();
     };
 
     sk_sp<SkSurface> surf = makeDirectBackendSurface();