diff --git a/ivy.xml b/ivy.xml
index 89cfdf6..f2fcade 100644
--- a/ivy.xml
+++ b/ivy.xml
@@ -16,7 +16,7 @@
 -->
 <!DOCTYPE ivy-module [
   <!ENTITY maven.version "2.0">
-  <!ENTITY gradle.version "3.2">
+  <!ENTITY gradle.version "3.4">
   <!ENTITY asm.version "9.2">
   <!ENTITY jarjar.asm.version "5.2">
 ]>
diff --git a/src/main/java/de/thetaphi/forbiddenapis/gradle/CheckForbiddenApis.java b/src/main/java/de/thetaphi/forbiddenapis/gradle/CheckForbiddenApis.java
index 1ff8293..f8244e1 100644
--- a/src/main/java/de/thetaphi/forbiddenapis/gradle/CheckForbiddenApis.java
+++ b/src/main/java/de/thetaphi/forbiddenapis/gradle/CheckForbiddenApis.java
@@ -39,7 +39,7 @@
 import org.gradle.api.file.FileTree;
 import org.gradle.api.file.FileTreeElement;
 import org.gradle.api.specs.Spec;
-import org.gradle.api.tasks.Classpath;
+import org.gradle.api.tasks.CompileClasspath;
 import org.gradle.api.tasks.Input;
 import org.gradle.api.tasks.InputFiles;
 import org.gradle.api.tasks.Internal;
@@ -148,8 +148,7 @@ public void setPatternSet(PatternSet patternSet) {
    * A {@link FileCollection} used to configure the classpath.
    * Defaults to current sourseSet's compile classpath.
    */
-  @InputFiles
-  @Classpath
+  @CompileClasspath
   public FileCollection getClasspath() {
     return classpath;
   }