diff --git a/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/GradleGenerator.kt b/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/GradleGenerator.kt
index 04427ec52b4..8d6d75b6bef 100644
--- a/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/GradleGenerator.kt
+++ b/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/GradleGenerator.kt
@@ -95,7 +95,6 @@ class GradleGenerator : KotlinIntegration {
     private fun generateSmokeTestJarTask(writer: GradleWriter, ctx: CodegenContext) {
         writer.withBlock("jvm {", "}") {
             withBlock("compilations {", "}") {
-                write("val runtimePath = configurations.getByName(#S).map { if (it.isDirectory) it else zipTree(it) }", "jvmRuntimeClasspath")
                 write("val mainPath = getByName(#S).output.classesDirs", "main")
                 write("val testPath = getByName(#S).output.classesDirs", "test")
                 withBlock("tasks {", "}") {
@@ -106,6 +105,7 @@ class GradleGenerator : KotlinIntegration {
                         withBlock("manifest {", "}") {
                             write("attributes[#S] = #S", "Main-Class", "${ctx.settings.pkg.name}.smoketests.SmokeTestsKt")
                         }
+                        write("val runtimePath = configurations.getByName(#S).map { if (it.isDirectory) it else zipTree(it) }", "jvmRuntimeClasspath")
                         write("duplicatesStrategy = DuplicatesStrategy.EXCLUDE")
                         write("from(runtimePath, mainPath, testPath)")
                         write("archiveBaseName.set(#S)", "\${project.name}-smoketests")