diff --git a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestInput.java b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestInput.java
index 17682401e1..3581e164fb 100644
--- a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestInput.java
+++ b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestInput.java
@@ -1,6 +1,7 @@
 package tech.picnic.errorprone.bugpatterns;
 
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 final class MethodTemplateTest implements RefasterTemplateTestCase {
   String testObjectReturn() {
diff --git a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestOutput.java b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestOutput.java
index 40347dab52..1793d44393 100644
--- a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestOutput.java
+++ b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/bugpatterns/MethodTemplatesTestOutput.java
@@ -1,6 +1,7 @@
 package tech.picnic.errorprone.bugpatterns;
 
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 final class MethodTemplateTest implements RefasterTemplateTestCase {
   String testObjectReturn() {