diff --git a/predict/lambda_function.py b/predict/lambda_function.py
index c37542c..b396ae0 100644
--- a/predict/lambda_function.py
+++ b/predict/lambda_function.py
@@ -3,6 +3,8 @@ import botocore.credentials
 from botocore.awsrequest import AWSRequest
 from botocore.endpoint import URLLib3Session
 from botocore.auth import SigV4Auth
+from aws_xray_sdk.core import xray_recorder
+from aws_xray_sdk.core import patch_all
 import json
 import os
 from datetime import datetime, timedelta, timezone
diff --git a/query/lambda_function.py b/query/lambda_function.py
index 3c9a9cd..61419cf 100644
--- a/query/lambda_function.py
+++ b/query/lambda_function.py
@@ -3,6 +3,8 @@ import botocore.credentials
 from botocore.awsrequest import AWSRequest
 from botocore.endpoint import URLLib3Session
 from botocore.auth import SigV4Auth
+from aws_xray_sdk.core import xray_recorder
+from aws_xray_sdk.core import patch_all
 import json
 import os
 from datetime import datetime, timedelta, timezone