diff --git a/buildSrc/src/main/kotlin/org/briarproject/briar/desktop/builddata/BuildDataPlugin.kt b/buildSrc/src/main/kotlin/org/briarproject/briar/desktop/builddata/BuildDataPlugin.kt
index 38d44dd34eade4ed98a09a621e2fb40b0b2d8b82..6afb72c4724894b9e95abd0f46f04c81a2de667d 100644
--- a/buildSrc/src/main/kotlin/org/briarproject/briar/desktop/builddata/BuildDataPlugin.kt
+++ b/buildSrc/src/main/kotlin/org/briarproject/briar/desktop/builddata/BuildDataPlugin.kt
@@ -3,7 +3,7 @@ package org.briarproject.briar.desktop.builddata
 import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.gradle.api.file.SourceDirectorySet
-import org.gradle.api.plugins.JavaPluginConvention
+import org.gradle.api.plugins.JavaPluginExtension
 
 class BuildDataPlugin : Plugin<Project> {
     override fun apply(project: Project) {
@@ -20,9 +20,7 @@ class BuildDataPlugin : Plugin<Project> {
         project.tasks.findByName("compileJava")!!.dependsOn(task)
         val pathBuildDir = project.buildDir.toPath()
         val source = Util.getSourceDir(pathBuildDir)
-        val sourceSets = project.convention
-            .getPlugin(JavaPluginConvention::class.java).sourceSets
-            .findByName("main")
+        val sourceSets = project.extensions.getByType(JavaPluginExtension::class.java).sourceSets.findByName("main")
         sourceSets!!.java { sourceSet: SourceDirectorySet -> sourceSet.srcDir(source) }
     }
 }