diff --git a/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java b/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java
index e9ad6115..b57ee5a3 100644
--- a/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java
+++ b/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java
@@ -6,6 +6,7 @@ import java.util.List;
 import graphql.GraphQLError;
 import graphql.GraphqlErrorBuilder;
 import graphql.schema.DataFetchingEnvironment;
+import org.springframework.util.Assert;
 import reactor.core.publisher.Mono;
 
 import org.springframework.graphql.execution.DataFetcherExceptionResolver;
@@ -40,6 +41,11 @@ public class SecurityDataFetcherExceptionResolver implements DataFetcherExceptio
 		return Mono.empty();
 	}
 
+	public void setAuthenticationTrustResolver(AuthenticationTrustResolver authenticationTrustResolver) {
+		Assert.notNull(authenticationTrustResolver, "authenticationTrustResolver cannot be null");
+		this.authenticationTrustResolver = authenticationTrustResolver;
+	}
+
 	private Mono<List<GraphQLError>> unauthorized(DataFetchingEnvironment environment) {
 		return Mono.fromCallable(() -> Arrays.asList(
 				GraphqlErrorBuilder.newError(environment)