diff --git a/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java b/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java
index c2f4f73..436ebb8 100644
--- a/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java
+++ b/src/tools/android/java/com/google/devtools/build/android/desugar/runtime/ConsumerWrapper.java
@@ -13,6 +13,8 @@
 // limitations under the License.
 package com.google.devtools.build.android.desugar.runtime;
 
+import android.accessibilityservice.AccessibilityService;
+import android.accessibilityservice.AccessibilityService.ScreenshotResult;
 import android.app.DirectAction;
 import android.os.Bundle;
 import android.os.CancellationSignal;
@@ -93,4 +95,15 @@
         resultExecutor,
         callback != null ? new ConsumerWrapper<List<DirectAction>>(callback) : null);
   }
+
+  public static boolean takeScreenshot(
+      AccessibilityService receiver,
+      int displayId,
+      Executor resultExecutor,
+      j$.util.function.Consumer<ScreenshotResult> callback) {
+    return receiver.takeScreenshot(
+        displayId,
+        resultExecutor,
+        callback != null ? new ConsumerWrapper<ScreenshotResult>(callback) : null);
+  }
 }