diff --git a/src/test/java/com/google/devtools/build/lib/buildtool/util/BuildIntegrationTestCase.java b/src/test/java/com/google/devtools/build/lib/buildtool/util/BuildIntegrationTestCase.java
index 43952d3..de22d13 100644
--- a/src/test/java/com/google/devtools/build/lib/buildtool/util/BuildIntegrationTestCase.java
+++ b/src/test/java/com/google/devtools/build/lib/buildtool/util/BuildIntegrationTestCase.java
@@ -60,7 +60,7 @@
 import com.google.devtools.build.lib.events.NullEventHandler;
 import com.google.devtools.build.lib.events.util.EventCollectionApparatus;
 import com.google.devtools.build.lib.exec.BinTools;
-import com.google.devtools.build.lib.exec.ExecutorBuilder;
+import com.google.devtools.build.lib.exec.ModuleActionContextRegistry;
 import com.google.devtools.build.lib.includescanning.IncludeScanningModule;
 import com.google.devtools.build.lib.integration.util.IntegrationMock;
 import com.google.devtools.build.lib.network.ConnectivityStatusProvider;
@@ -304,9 +304,11 @@
       }
 
       @Override
-      public void executorInit(
-          CommandEnvironment env, BuildRequest request, ExecutorBuilder builder) {
-        builder.addActionContext(
+      public void registerActionContexts(
+          ModuleActionContextRegistry.Builder registryBuilder,
+          CommandEnvironment env,
+          BuildRequest buildRequest) {
+        registryBuilder.register(
             WorkspaceStatusAction.Context.class, new DummyWorkspaceStatusActionContext());
       }
     };