diff --git a/examples/data_masking/src/getting_started_encrypt_data.py b/examples/data_masking/src/getting_started_encrypt_data.py
index a6de0954840..795b22b9a00 100644
--- a/examples/data_masking/src/getting_started_encrypt_data.py
+++ b/examples/data_masking/src/getting_started_encrypt_data.py
@@ -18,7 +18,7 @@
 
 
 @logger.inject_lambda_context
-def lambda_handler(event: dict, context: LambdaContext) -> str:
+def lambda_handler(event: dict, context: LambdaContext) -> dict:
     data: dict = event.get("body", {})
 
     logger.info("Encrypting the whole object")