diff --git a/simpleexample/build.gradle b/simpleexample/build.gradle
index 5a0e5c39d2bc627bff0c90e90d97dedb3c62898b..58ca1213a6dc59afb4f3c61a46f72741becc8195 100644
--- a/simpleexample/build.gradle
+++ b/simpleexample/build.gradle
@@ -26,19 +26,17 @@ javafx {
     // launch debugger using Remote Java Application debug launch configuration
 }
 
-test {
-	// don't run UI test on server (that cannot run javafx)
-	if (project.hasProperty('gitlab-ci')) {
-		exclude 'simpleex.ui/*'
-	}
+sourceSets {
+    main {
+        java {
+			// don't include UI without FxMapControl installed
+			if (project.hasProperty('gitlab-ci')) {
+                exclude 'simpleex.ui/*'
+            }
+        }
+    }
 }
 
-compile {
-	// don't compile UI without FxMapControl installed
-	if (project.hasProperty('gitlab-ci')) {
-		exclude 'simpleex.ui/*'
-	}
-}
 test {
 	// don't run UI test on server (that cannot run javafx)
 	if (project.hasProperty('gitlab-ci')) {