diff --git a/tests/e2e/idempotency_redis/handlers/function_thread_safety_handler.py b/tests/e2e/idempotency_redis/handlers/function_thread_safety_handler.py
index d64ff5df10..6d326c17e4 100644
--- a/tests/e2e/idempotency_redis/handlers/function_thread_safety_handler.py
+++ b/tests/e2e/idempotency_redis/handlers/function_thread_safety_handler.py
@@ -4,9 +4,9 @@
 from threading import current_thread
 
 from aws_lambda_powertools.utilities.idempotency import (
-    RedisCachePersistenceLayer,
     idempotent_function,
 )
+from aws_lambda_powertools.utilities.idempotency.persistence.redis import RedisCachePersistenceLayer
 
 REDIS_HOST = os.getenv("RedisEndpoint", "")
 persistence_layer = RedisCachePersistenceLayer(host=REDIS_HOST, port=6379)
diff --git a/tests/e2e/idempotency_redis/handlers/optional_idempotency_key_handler.py b/tests/e2e/idempotency_redis/handlers/optional_idempotency_key_handler.py
index 4182812fc4..75be8b3129 100644
--- a/tests/e2e/idempotency_redis/handlers/optional_idempotency_key_handler.py
+++ b/tests/e2e/idempotency_redis/handlers/optional_idempotency_key_handler.py
@@ -3,9 +3,9 @@
 
 from aws_lambda_powertools.utilities.idempotency import (
     IdempotencyConfig,
-    RedisCachePersistenceLayer,
     idempotent,
 )
+from aws_lambda_powertools.utilities.idempotency.persistence.redis import RedisCachePersistenceLayer
 
 REDIS_HOST = os.getenv("RedisEndpoint", "")
 persistence_layer = RedisCachePersistenceLayer(host=REDIS_HOST, port=6379)
diff --git a/tests/e2e/idempotency_redis/handlers/parallel_execution_handler.py b/tests/e2e/idempotency_redis/handlers/parallel_execution_handler.py
index 5dd4fbdcbd..c28f84f746 100644
--- a/tests/e2e/idempotency_redis/handlers/parallel_execution_handler.py
+++ b/tests/e2e/idempotency_redis/handlers/parallel_execution_handler.py
@@ -2,9 +2,9 @@
 import time
 
 from aws_lambda_powertools.utilities.idempotency import (
-    RedisCachePersistenceLayer,
     idempotent,
 )
+from aws_lambda_powertools.utilities.idempotency.persistence.redis import RedisCachePersistenceLayer
 
 REDIS_HOST = os.getenv("RedisEndpoint", "")
 persistence_layer = RedisCachePersistenceLayer(host=REDIS_HOST, port=6379)
diff --git a/tests/e2e/idempotency_redis/handlers/ttl_cache_expiration_handler.py b/tests/e2e/idempotency_redis/handlers/ttl_cache_expiration_handler.py
index 2f938d3c46..a93413e157 100644
--- a/tests/e2e/idempotency_redis/handlers/ttl_cache_expiration_handler.py
+++ b/tests/e2e/idempotency_redis/handlers/ttl_cache_expiration_handler.py
@@ -3,9 +3,9 @@
 
 from aws_lambda_powertools.utilities.idempotency import (
     IdempotencyConfig,
-    RedisCachePersistenceLayer,
     idempotent,
 )
+from aws_lambda_powertools.utilities.idempotency.persistence.redis import RedisCachePersistenceLayer
 
 REDIS_HOST = os.getenv("RedisEndpoint", "")
 persistence_layer = RedisCachePersistenceLayer(host=REDIS_HOST, port=6379)
diff --git a/tests/e2e/idempotency_redis/handlers/ttl_cache_timeout_handler.py b/tests/e2e/idempotency_redis/handlers/ttl_cache_timeout_handler.py
index 3d8ab335c8..56c2372b2c 100644
--- a/tests/e2e/idempotency_redis/handlers/ttl_cache_timeout_handler.py
+++ b/tests/e2e/idempotency_redis/handlers/ttl_cache_timeout_handler.py
@@ -3,9 +3,9 @@
 
 from aws_lambda_powertools.utilities.idempotency import (
     IdempotencyConfig,
-    RedisCachePersistenceLayer,
     idempotent,
 )
+from aws_lambda_powertools.utilities.idempotency.persistence.redis import RedisCachePersistenceLayer
 
 REDIS_HOST = os.getenv("RedisEndpoint", "")
 persistence_layer = RedisCachePersistenceLayer(host=REDIS_HOST, port=6379)