diff --git a/build.gradle b/build.gradle
index 63e5bf26..72b87431 100644
--- a/build.gradle
+++ b/build.gradle
@@ -211,7 +211,8 @@ task collectScreenshots {
                                 def mainMethod = klass.getMethod("main")
                                 println "Collecting screenshot for ${klassName}"
                                 javaexec {
-                                    classpath set.runtimeClasspath
+                                    classpath += set.runtimeClasspath
+                                    classpath += set.compileClasspath
                                     def className = y.name.replace(".class", "")
                                     main = className
                                     jvmArgs += "-DtakeScreenshot=true"