diff --git a/src/main/groovy/no/nils/wsdl2java/Wsdl2JavaPlugin.groovy b/src/main/groovy/no/nils/wsdl2java/Wsdl2JavaPlugin.groovy
index c0d705d..a43d9d9 100644
--- a/src/main/groovy/no/nils/wsdl2java/Wsdl2JavaPlugin.groovy
+++ b/src/main/groovy/no/nils/wsdl2java/Wsdl2JavaPlugin.groovy
@@ -27,7 +27,7 @@ class Wsdl2JavaPlugin implements Plugin<Project> {
         def wsdl2javaConfiguration = project.configurations.maybeCreate(WSDL2JAVA)
 
         // Get compile configuration and add Java 9+ dependencies if required.
-        project.configurations.named("compile").configure {
+        project.configurations.named("implementation").configure {
             it.withDependencies {
                 if (JavaVersion.current().isJava9Compatible()) {
                     JAVA_9_DEPENDENCIES.each { dep -> it.add(project.dependencies.create(dep)) }