diff --git a/src/commonMain/kotlin/wizard/files/composeApp/ModuleBuildGradleKts.kt b/src/commonMain/kotlin/wizard/files/composeApp/ModuleBuildGradleKts.kt
index 4f08800..7e93e94 100644
--- a/src/commonMain/kotlin/wizard/files/composeApp/ModuleBuildGradleKts.kt
+++ b/src/commonMain/kotlin/wizard/files/composeApp/ModuleBuildGradleKts.kt
@@ -50,6 +50,7 @@ class ModuleBuildGradleKts(info: ProjectInfo) : ProjectFile {
         appendLine("}")
         appendLine("")
         appendLine("kotlin {")
+        appendLine("    task(\"testClasses\")")
         if (info.hasPlatform(ProjectPlatform.Android)) {
             appendLine("    androidTarget {")
             appendLine("        compilations.all {")
diff --git a/src/commonMain/kotlin/wizard/files/kmpLibrary/ModuleBuildGradleKts.kt b/src/commonMain/kotlin/wizard/files/kmpLibrary/ModuleBuildGradleKts.kt
index 1a7ef08..5702ece 100644
--- a/src/commonMain/kotlin/wizard/files/kmpLibrary/ModuleBuildGradleKts.kt
+++ b/src/commonMain/kotlin/wizard/files/kmpLibrary/ModuleBuildGradleKts.kt
@@ -28,6 +28,7 @@ class ModuleBuildGradleKts(info: ProjectInfo) : ProjectFile {
         appendLine("version = \"1.0\"")
         appendLine("")
         appendLine("kotlin {")
+        appendLine("    task(\"testClasses\")")
         if (info.hasPlatform(ProjectPlatform.Android)) {
             appendLine("    androidTarget {")
             appendLine("        publishLibraryVariants(\"release\")")
diff --git a/src/jvmTest/kotlin/wizard/ComposeAppGeneratorTest.kt b/src/jvmTest/kotlin/wizard/ComposeAppGeneratorTest.kt
index a6300eb..dd65229 100644
--- a/src/jvmTest/kotlin/wizard/ComposeAppGeneratorTest.kt
+++ b/src/jvmTest/kotlin/wizard/ComposeAppGeneratorTest.kt
@@ -93,6 +93,7 @@ class ComposeAppGeneratorTest {
                 }
 
                 kotlin {
+                    task("testClasses")
                     androidTarget {
                         compilations.all {
                             compileTaskProvider {
@@ -405,6 +406,7 @@ class ComposeAppGeneratorTest {
                 }
 
                 kotlin {
+                    task("testClasses")
                     androidTarget {
                         compilations.all {
                             compileTaskProvider {
@@ -543,6 +545,7 @@ class ComposeAppGeneratorTest {
                 }
 
                 kotlin {
+                    task("testClasses")
                     listOf(
                         iosX64(),
                         iosArm64(),
@@ -629,6 +632,7 @@ class ComposeAppGeneratorTest {
                 }
 
                 kotlin {
+                    task("testClasses")
                     jvm()
 
                     sourceSets {
@@ -719,6 +723,7 @@ class ComposeAppGeneratorTest {
                 }
 
                 kotlin {
+                    task("testClasses")
                     js {
                         browser()
                         binaries.executable()
@@ -800,6 +805,7 @@ class ComposeAppGeneratorTest {
                 }
 
                 kotlin {
+                    task("testClasses")
                     wasmJs {
                         browser()
                         binaries.executable()
diff --git a/src/jvmTest/kotlin/wizard/KmpLibraryGeneratorTest.kt b/src/jvmTest/kotlin/wizard/KmpLibraryGeneratorTest.kt
index b6e5b97..6c403ce 100644
--- a/src/jvmTest/kotlin/wizard/KmpLibraryGeneratorTest.kt
+++ b/src/jvmTest/kotlin/wizard/KmpLibraryGeneratorTest.kt
@@ -56,6 +56,7 @@ class GeneratorTest {
                 version = "1.0"
 
                 kotlin {
+                    task("testClasses")
                     androidTarget {
                         publishLibraryVariants("release")
                         compilations.all {
@@ -368,6 +369,7 @@ class GeneratorTest {
                 version = "1.0"
 
                 kotlin {
+                    task("testClasses")
                     jvm()
 
                     sourceSets {
@@ -429,6 +431,7 @@ class GeneratorTest {
                 version = "1.0"
                 
                 kotlin {
+                    task("testClasses")
                     androidTarget {
                         publishLibraryVariants("release")
                         compilations.all {