diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py index a2af8846..c85d8bb4 100644 --- a/tests/test_wrapper.py +++ b/tests/test_wrapper.py @@ -31,7 +31,7 @@ class TestDatadogLambdaWrapper(unittest.TestCase): def setUp(self): # Force @datadog_lambda_wrapper to always create a real # (not no-op) wrapper. - patch("ddtrace.internal.remoteconfig.RemoteConfig").start() + patch("ddtrace.internal.remoteconfig.worker.RemoteConfigPoller").start() patch("ddtrace.internal.writer.AgentWriter.flush_queue").start() wrapper.datadog_lambda_wrapper._force_wrap = True