diff --git a/packages/react-native-codegen/android/gradlePlugin-build/gradlePlugin/src/main/java/com/facebook/react/codegen/plugin/CodegenPlugin.java b/packages/react-native-codegen/android/gradlePlugin-build/gradlePlugin/src/main/java/com/facebook/react/codegen/plugin/CodegenPlugin.java
index 055bcff1b6f..4ea0a13e09f 100644
--- a/packages/react-native-codegen/android/gradlePlugin-build/gradlePlugin/src/main/java/com/facebook/react/codegen/plugin/CodegenPlugin.java
+++ b/packages/react-native-codegen/android/gradlePlugin-build/gradlePlugin/src/main/java/com/facebook/react/codegen/plugin/CodegenPlugin.java
@@ -98,10 +98,13 @@ public class CodegenPlugin implements Plugin<Project> {
               task.getOutputs().dir(outputDir);
 
               if (extension.useJavaGenerator) {
-                generateJavaFromSchemaWithJavaGenerator(
-                    generatedSchemaFile,
-                    extension.codegenJavaPackageName,
-                    new File(generatedSrcDir, "java"));
+                task.doLast(
+                    s -> {
+                      generateJavaFromSchemaWithJavaGenerator(
+                          generatedSchemaFile,
+                          extension.codegenJavaPackageName,
+                          new File(generatedSrcDir, "java"));
+                    });
                 // TODO: generate JNI C++ files.
                 task.commandLine("echo");
               } else {