diff --git a/src/main/kotlin/org/springdoc/openapi/gradle/plugin/Constants.kt b/src/main/kotlin/org/springdoc/openapi/gradle/plugin/Constants.kt
index 95dae65..e9d8c0f 100644
--- a/src/main/kotlin/org/springdoc/openapi/gradle/plugin/Constants.kt
+++ b/src/main/kotlin/org/springdoc/openapi/gradle/plugin/Constants.kt
@@ -4,6 +4,8 @@ const val EXTENSION_NAME = "openApi"
 const val GROUP_NAME = "OpenApi"
 const val OPEN_API_TASK_NAME = "generateOpenApiDocs"
 const val OPEN_API_TASK_DESCRIPTION = "Generates the spring doc openapi file"
+
+const val JAVA_ASSEMBLE_TASK_NAME = "assemble"
 const val SPRING_BOOT_RUN_TASK_NAME = "bootRun"
 const val SPRING_BOOT_RUN_MAIN_CLASS_NAME_TASK_NAME = "bootRunMainClassName"
 const val SPRING_BOOT_3_RUN_MAIN_CLASS_NAME_TASK_NAME = "bootRun"
diff --git a/src/main/kotlin/org/springdoc/openapi/gradle/plugin/OpenApiGradlePlugin.kt b/src/main/kotlin/org/springdoc/openapi/gradle/plugin/OpenApiGradlePlugin.kt
index 64ca896..fe7e34e 100644
--- a/src/main/kotlin/org/springdoc/openapi/gradle/plugin/OpenApiGradlePlugin.kt
+++ b/src/main/kotlin/org/springdoc/openapi/gradle/plugin/OpenApiGradlePlugin.kt
@@ -18,7 +18,9 @@ open class OpenApiGradlePlugin : Plugin<Project> {
 			plugins.apply(EXEC_FORK_PLUGIN)
 
 			extensions.create(EXTENSION_NAME, OpenApiExtension::class.java)
-			tasks.register(FORKED_SPRING_BOOT_RUN_TASK_NAME, JavaExecFork::class.java)
+			tasks.register(FORKED_SPRING_BOOT_RUN_TASK_NAME, JavaExecFork::class.java) { task ->
+				task.dependsOn(JAVA_ASSEMBLE_TASK_NAME)
+			}
 			tasks.register(OPEN_API_TASK_NAME, OpenApiGeneratorTask::class.java)
 
 			generate(this)
@@ -55,6 +57,7 @@ open class OpenApiGradlePlugin : Plugin<Project> {
 			FORKED_SPRING_BOOT_RUN_TASK_NAME,
 			JavaExecFork::class.java
 		) { fork ->
+			fork.dependsOn(tasks.named(JAVA_ASSEMBLE_TASK_NAME))
 			fork.dependsOn(tasks.named(bootRunMainClassNameTask.name))
 			fork.onlyIf { needToFork(bootRunTask, customBootRun, fork) }
 		}