diff --git a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/AssertJOptionalRules.java b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/AssertJOptionalRules.java
index 9cdb31a3..561ae7b0 100644
--- a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/AssertJOptionalRules.java
+++ b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/AssertJOptionalRules.java
@@ -12,7 +12,6 @@ import java.util.function.Predicate;
 import org.assertj.core.api.AbstractAssert;
 import org.assertj.core.api.AbstractObjectAssert;
 import org.assertj.core.api.AbstractOptionalAssert;
-import org.assertj.core.api.AbstractStringAssert;
 import org.assertj.core.api.ObjectAssert;
 import org.assertj.core.api.OptionalAssert;
 import tech.picnic.errorprone.refaster.annotation.OnlineDocumentation;
@@ -106,7 +105,7 @@ final class AssertJOptionalRules {
 
   static final class AssertThatHasValue<T> {
     @BeforeTemplate
-    AbstractStringAssert<?> before(Optional<String> optional, String value) {
+    ObjectAssert<T> before(Optional<T> optional, T value) {
       return assertThat(optional.orElseThrow()).isEqualTo(value);
     }