import static org.mockserver.matchers.Times.exactly;
 import static org.mockserver.model.HttpForward.forward;
 import static org.mockserver.model.HttpRequest.request;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.ApplicationContext;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.http.MediaType;
 import org.springframework.test.web.reactive.server.WebTestClient;
-import org.springframework.web.context.WebApplicationContext;
 import org.springframework.web.reactive.function.client.WebClient;
 
 @SpringBootTest
 class DemoApplicationIT extends IntegrationTestBase
 {
   WebTestClient webClient;
+  @Autowired
+  ApplicationContext context;
 
 
   @BeforeEach
-  void setUp(WebApplicationContext context)
+  void setUp()
   {
     webClient = WebTestClient.bindToApplicationContext(context).build();
   }