diff --git a/error-prone-contrib/pom.xml b/error-prone-contrib/pom.xml
index cd4016dfe84..d04e1b5ef5c 100644
--- a/error-prone-contrib/pom.xml
+++ b/error-prone-contrib/pom.xml
@@ -181,6 +181,11 @@
             <artifactId>mongodb-driver-core</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <scope>test</scope>
+        </dependency>
         <dependency>
             <groupId>org.reactivestreams</groupId>
             <artifactId>reactive-streams</artifactId>
diff --git a/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/DirectReturnTest.java b/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/DirectReturnTest.java
index f5d3e15298c..19af8b4a34c 100644
--- a/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/DirectReturnTest.java
+++ b/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/DirectReturnTest.java
@@ -226,18 +226,8 @@ void replacement() {
 
   @Test
   void ignoreClassesAnnotatedWithLombokData() {
-    BugCheckerRefactoringTestHelper.newInstance(DirectReturn.class, getClass())
-        .addInputLines("Data.java", "package lombok;", "", "public @interface Data {}")
-        .expectUnchanged()
-        .addInputLines(
-            "A.java",
-            "import lombok.Data;",
-            "",
-            "@Data",
-            "class A {",
-            "  private String field;",
-            "}")
-        .addOutputLines(
+    CompilationTestHelper.newInstance(DirectReturn.class, getClass())
+        .addSourceLines(
             "A.java",
             "import lombok.Data;",
             "",
@@ -245,6 +235,7 @@ void ignoreClassesAnnotatedWithLombokData() {
             "class A {",
             "  private String field;",
             "}")
-        .doTest(TestMode.TEXT_MATCH);
+        .expectNoDiagnostics()
+        .doTest();
   }
 }
diff --git a/pom.xml b/pom.xml
index a474e3567fa..d08c940a5da 100644
--- a/pom.xml
+++ b/pom.xml
@@ -456,6 +456,11 @@
                     </exclusion>
                 </exclusions>
             </dependency>
+            <dependency>
+                <groupId>org.projectlombok</groupId>
+                <artifactId>lombok</artifactId>
+                <version>1.18.28</version>
+            </dependency>
             <dependency>
                 <groupId>org.slf4j</groupId>
                 <artifactId>slf4j-api</artifactId>