diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py
index 35b5b9fa..87edd322 100644
--- a/tests/test_wrapper.py
+++ b/tests/test_wrapper.py
@@ -16,10 +16,12 @@
 from datadog_lambda.metric import lambda_metric
 from datadog_lambda.thread_stats_writer import ThreadStatsWriter
 from ddtrace.trace import Span, tracer
-from ddtrace.internal.constants import MAX_UINT_64BITS
+
 
 from tests.utils import get_mock_context, reset_xray_connection
 
+MAX_UINT_64BITS = (1 << 64) - 1
+
 
 class TestDatadogLambdaWrapper(unittest.TestCase):
     def setUp(self):