diff --git a/buildSrc/src/main/kotlin/CollectScreenShots.kt b/buildSrc/src/main/kotlin/CollectScreenShots.kt
index 35a9f3b1..0413f542 100644
--- a/buildSrc/src/main/kotlin/CollectScreenShots.kt
+++ b/buildSrc/src/main/kotlin/CollectScreenShots.kt
@@ -115,7 +115,7 @@ object ScreenshotsHelper {
     ): CollectScreenshotsTask {
         val task = project.tasks.register<CollectScreenshotsTask>("collectScreenshots").get()
         task.outputDir.set(project.file(project.projectDir.toString() + "/images"))
-        task.inputDir.set(File(project.buildDir, "classes/kotlin/${sourceSet.name}"))
+        task.inputDir.set(File(project.layout.buildDirectory.get().asFile, "classes/kotlin/${sourceSet.name}"))
         task.runtimeDependencies.set(sourceSet.runtimeClasspath)
         task.config()
         task.dependsOn(sourceSet.output)
diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties
index ac72c34e..3fa8f862 100644
--- a/gradle/wrapper/gradle-wrapper.properties
+++ b/gradle/wrapper/gradle-wrapper.properties
@@ -1,6 +1,6 @@
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
-distributionUrl=https\://services.gradle.org/distributions/gradle-8.3-bin.zip
+distributionUrl=https\://services.gradle.org/distributions/gradle-8.4-bin.zip
 networkTimeout=10000
 validateDistributionUrl=true
 zipStoreBase=GRADLE_USER_HOME