diff --git a/smoke-tests/build.gradle.kts b/smoke-tests/build.gradle.kts
index f29e2cb47634..f1719c82abcd 100644
--- a/smoke-tests/build.gradle.kts
+++ b/smoke-tests/build.gradle.kts
@@ -72,13 +72,14 @@ tasks {
       }
     }
 
-    val shadowTask = project(":javaagent").tasks.named<ShadowJar>("shadowJar").get()
-    inputs.files(layout.files(shadowTask))
+    val shadowTask = project(":javaagent").tasks.named<ShadowJar>("shadowJar")
+    val agentJarPath = shadowTask.flatMap { it.archiveFile }
+    inputs.files(agentJarPath)
       .withPropertyName("javaagent")
       .withNormalizer(ClasspathNormalizer::class)
 
     doFirst {
-      jvmArgs("-Dio.opentelemetry.smoketest.agent.shadowJar.path=${shadowTask.archiveFile.get()}")
+      jvmArgs("-Dio.opentelemetry.smoketest.agent.shadowJar.path=${agentJarPath.get()}")
     }
   }
 }