diff --git a/docs/utilities/idempotency.md b/docs/utilities/idempotency.md
index e448d82e28e..12baabd047e 100644
--- a/docs/utilities/idempotency.md
+++ b/docs/utilities/idempotency.md
@@ -950,7 +950,7 @@ You can set up a `response_hook` in the `IdempotentConfig` class to manipulate t
 
 === "Using an Idempotent Response Hook"
 
-    ```python hl_lines="18 20 23 32"
+    ```python hl_lines="19 21 27 34"
     --8<-- "examples/idempotency/src/working_with_response_hook.py"
     ```
 
diff --git a/examples/idempotency/src/working_with_response_hook.py b/examples/idempotency/src/working_with_response_hook.py
index 725a56f32ba..2c2208d25a5 100644
--- a/examples/idempotency/src/working_with_response_hook.py
+++ b/examples/idempotency/src/working_with_response_hook.py
@@ -8,7 +8,7 @@
     IdempotencyConfig,
     idempotent_function,
 )
-from aws_lambda_powertools.utilities.idempotency.persistence.base import (
+from aws_lambda_powertools.utilities.idempotency.persistence.datarecord import (
     DataRecord,
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext