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 319481b241c..f5d3e15298c 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
@@ -227,7 +227,7 @@ void replacement() {
   @Test
   void ignoreClassesAnnotatedWithLombokData() {
     BugCheckerRefactoringTestHelper.newInstance(DirectReturn.class, getClass())
-        .addInputLines("Data.java", "package lombok;", "public @interface Data {}")
+        .addInputLines("Data.java", "package lombok;", "", "public @interface Data {}")
         .expectUnchanged()
         .addInputLines(
             "A.java",
diff --git a/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/util/MoreMatchersTest.java b/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/util/MoreMatchersTest.java
index f0b812ed719..dbf93124918 100644
--- a/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/util/MoreMatchersTest.java
+++ b/error-prone-contrib/src/test/java/tech/picnic/errorprone/bugpatterns/util/MoreMatchersTest.java
@@ -53,7 +53,7 @@ void hasMetaAnnotation() {
   @Test
   void hasLombokDataAnnotation() {
     CompilationTestHelper.newInstance(LombokDataAnnotationMatcher.class, getClass())
-        .addSourceLines("Data.java", "package lombok;", "public @interface Data {}")
+        .addSourceLines("Data.java", "package lombok;", "", "public @interface Data {}")
         .addSourceLines(
             "A.java",
             "import lombok.Data;",
@@ -63,17 +63,17 @@ void hasLombokDataAnnotation() {
             "public class A {",
             "  private String field;",
             "",
-            "  static class B { }",
+            "  static class B {}",
             "",
             "  @Data",
             "  // BUG: Diagnostic contains:",
-            "  static class C { }",
+            "  static class C {}",
             "}")
-        .addSourceLines("D.java", "import lombok.Data;", "", "public class D { }")
+        .addSourceLines("D.java", "public class D {}")
         .doTest();
   }
 
-  /** A {@link BugChecker} that delegates to {@link MoreMatchers#hasMetaAnnotation(String)} . */
+  /** A {@link BugChecker} that delegates to {@link MoreMatchers#hasMetaAnnotation(String)}. */
   @BugPattern(summary = "Interacts with `MoreMatchers` for testing purposes", severity = ERROR)
   public static final class TestTemplateMatcher extends BugChecker
       implements AnnotationTreeMatcher {
@@ -87,7 +87,7 @@ public Description matchAnnotation(AnnotationTree tree, VisitorState state) {
     }
   }
 
-  /** A {@link BugChecker} that delegates to {@link MoreMatchers#HAS_LOMBOK_DATA} . */
+  /** A {@link BugChecker} that delegates to {@link MoreMatchers#HAS_LOMBOK_DATA}. */
   @BugPattern(summary = "Interacts with `MoreMatchers` for testing purposes", severity = ERROR)
   public static final class LombokDataAnnotationMatcher extends BugChecker
       implements ClassTreeMatcher {