diff --git a/build.gradle b/build.gradle
index 0aa7782..97f3650 100644
--- a/build.gradle
+++ b/build.gradle
@@ -13,9 +13,11 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+plugins {
+  id 'java'
+}
 
 apply from: "https://raw.githubusercontent.com/gocd/gocd-plugin-gradle-task-helpers/master/helper.gradle?_=${(int) (new Date().toInstant().epochSecond / 60)}"
-apply plugin: 'java'
 
 gocdPlugin {
   id = 'com.thoughtworks.gocd.elastic-agent.azure'
@@ -43,8 +45,11 @@ gocdPlugin {
 }
 
 version = gocdPlugin.fullVersion(project)
-sourceCompatibility = 11
-targetCompatibility = 11
+
+java {
+  sourceCompatibility = JavaVersion.VERSION_11
+  targetCompatibility = JavaVersion.VERSION_11
+}
 
 // In this section you declare where to find the dependencies of your project
 repositories {