diff --git a/src/test/java/com/example/demo/ReviewSubscriptionTest.java b/src/test/java/com/example/demo/ReviewSubscriptionTest.java
index 92f6d89..fb8e631 100644
--- a/src/test/java/com/example/demo/ReviewSubscriptionTest.java
+++ b/src/test/java/com/example/demo/ReviewSubscriptionTest.java
@@ -10,9 +10,9 @@
 import com.example.demo.services.ShowsService;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.netflix.graphql.dgs.DgsQueryExecutor;
-import com.netflix.graphql.dgs.autoconfig.DgsAutoConfiguration;
 import com.netflix.graphql.dgs.client.codegen.GraphQLQueryRequest;
 import com.netflix.graphql.dgs.scalars.UploadScalar;
+import com.netflix.graphql.dgs.test.EnableDgsTest;
 import graphql.ExecutionResult;
 import org.junit.jupiter.api.Test;
 import org.reactivestreams.Publisher;
@@ -34,7 +34,8 @@
  * Each time a review is added, a new ExecutionResult is given to subscriber.
  * Normally, this publisher is consumed by the Websocket/SSE subscription handler and you don't deal with this code directly, but for testing purposes it's useful to use the stream directly.
  */
-@SpringBootTest(classes = {DefaultReviewsService.class, ReviewsDataFetcher.class, DgsAutoConfiguration.class, DateTimeScalar.class, UploadScalar.class})
+@SpringBootTest(classes = {DefaultReviewsService.class, ReviewsDataFetcher.class, DateTimeScalar.class, UploadScalar.class})
+@EnableDgsTest
 public class ReviewSubscriptionTest {
     @Autowired
     DgsQueryExecutor dgsQueryExecutor;
diff --git a/src/test/java/com/example/demo/ShowsDataFetcherTest.java b/src/test/java/com/example/demo/ShowsDataFetcherTest.java
index 4dbb611..ff74164 100644
--- a/src/test/java/com/example/demo/ShowsDataFetcherTest.java
+++ b/src/test/java/com/example/demo/ShowsDataFetcherTest.java
@@ -13,7 +13,6 @@
 import com.example.demo.services.ShowsService;
 import com.jayway.jsonpath.TypeRef;
 import com.netflix.graphql.dgs.DgsQueryExecutor;
-import com.netflix.graphql.dgs.autoconfig.DgsAutoConfiguration;
 import com.netflix.graphql.dgs.client.codegen.GraphQLQueryRequest;
 import com.netflix.graphql.dgs.scalars.UploadScalar;
 import com.netflix.graphql.dgs.test.EnableDgsTest;