diff --git a/src/test/apiUsageSample/com/android/tools/apiusagesample/R8ApiUsageSample.java b/src/test/apiUsageSample/com/android/tools/apiusagesample/R8ApiUsageSample.java
index 8f57473..2061722 100644
--- a/src/test/apiUsageSample/com/android/tools/apiusagesample/R8ApiUsageSample.java
+++ b/src/test/apiUsageSample/com/android/tools/apiusagesample/R8ApiUsageSample.java
@@ -409,8 +409,9 @@
     try {
       R8.run(
           R8Command.builder(handler)
+              .setDisableTreeShaking(true)
               .setMinApiLevel(minApiLevel)
-              .setProgramConsumer(new R8ApiUsageSample.EnsureOutputConsumer())
+              .setProgramConsumer(new EnsureOutputConsumer())
               .addLibraryFiles(libraries)
               .addProgramFiles(inputs)
               .addAssertionsConfiguration(b -> b.setScopeAll().setEnable().build())