From 91d6f4a05e8157994adca9b47a9affc370f755ca Mon Sep 17 00:00:00 2001
From: Rick Ossendrijver <rick.ossies@hotmail.com>
Date: Mon, 17 Jan 2022 11:13:58 +0100
Subject: [PATCH] AssertJ: add missing `ImportPolicy` to after template

---
 .../picnic/errorprone/refastertemplates/AssertJTemplates.java    | 1 +
 1 file changed, 1 insertion(+)

diff --git a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refastertemplates/AssertJTemplates.java b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refastertemplates/AssertJTemplates.java
index 0458fc05ff..82913fc248 100644
--- a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refastertemplates/AssertJTemplates.java
+++ b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refastertemplates/AssertJTemplates.java
@@ -365,6 +365,7 @@ ObjectEnumerableAssert<?, S> before(Set<S> set, T element) {
     }
 
     @AfterTemplate
+    @UseImportPolicy(ImportPolicy.STATIC_IMPORT_ALWAYS)
     ObjectEnumerableAssert<?, S> after(Set<S> set, T element) {
       return assertThat(set).containsExactly(element);
     }