diff --git a/api-gateway/src/test/java/com/example/api/gateway/TestAPIGatewayApplication.java b/api-gateway/src/test/java/com/example/api/gateway/TestAPIGatewayApplication.java
index 149d479d..56055371 100644
--- a/api-gateway/src/test/java/com/example/api/gateway/TestAPIGatewayApplication.java
+++ b/api-gateway/src/test/java/com/example/api/gateway/TestAPIGatewayApplication.java
@@ -27,9 +27,8 @@ GenericContainer<?> zipkinContainer() {
 
     @Bean
     @ServiceConnection(name = "redis")
-    GenericContainer<?> redisContainer() {
-        return new GenericContainer<>(DockerImageName.parse("redis").withTag("7.4.0-alpine"))
-                .withExposedPorts(6379)
+    RedisContainer redisContainer() {
+        return new RedisContainer(DockerImageName.parse("redis").withTag("7.4.0-alpine"))
                 .withReuse(true);
     }