diff --git a/spring-cloud-contract-tools/spring-cloud-contract-gradle-plugin/src/main/groovy/org/springframework/cloud/contract/verifier/plugin/GenerateServerTestsTask.groovy b/spring-cloud-contract-tools/spring-cloud-contract-gradle-plugin/src/main/groovy/org/springframework/cloud/contract/verifier/plugin/GenerateServerTestsTask.groovy
index 016fbe4b34..71f88f767d 100644
--- a/spring-cloud-contract-tools/spring-cloud-contract-gradle-plugin/src/main/groovy/org/springframework/cloud/contract/verifier/plugin/GenerateServerTestsTask.groovy
+++ b/spring-cloud-contract-tools/spring-cloud-contract-gradle-plugin/src/main/groovy/org/springframework/cloud/contract/verifier/plugin/GenerateServerTestsTask.groovy
@@ -55,7 +55,10 @@ class GenerateServerTestsTask extends ConventionTask {
 		project.logger.info("Contracts are unpacked to [${contractsDslDir}]")
 		project.logger.info("Included contracts are [${props.includedContracts}]")
 
-		project.sourceSets.test.groovy {
+		def sourceSetType = getConfigProperties().getTestFramework() == TestFramework.SPOCK ?
+				"groovy" : "java"
+
+		project.sourceSets.test."${sourceSetType}" {
 			project.logger.
 					info("Registering ${getConfigProperties().generatedTestSourcesDir} as test source directory")
 			srcDir getConfigProperties().getGeneratedTestSourcesDir()