diff --git a/tests/tracer/test_memory_leak.py b/tests/tracer/test_memory_leak.py
index 2963268fd65..b13cc0ec4ee 100644
--- a/tests/tracer/test_memory_leak.py
+++ b/tests/tracer/test_memory_leak.py
@@ -1,26 +1,21 @@
 """
 Variety of test cases ensuring that ddtrace does not leak memory.
 """
-from typing import TYPE_CHECKING
 from weakref import WeakValueDictionary
 
 import pytest
 
+from ddtrace.trace import Span
+from ddtrace.trace import Tracer
 from tests.utils import DummyTracer
 
 
-if TYPE_CHECKING:  # pragma: no cover
-    from ddtrace.trace import Span  # noqa:F401
-    from ddtrace.trace import Tracer  # noqa:F401
-
-
 @pytest.fixture
 def tracer() -> DummyTracer:
     return DummyTracer()
 
 
-def trace(weakdict: WeakValueDictionary, tracer: Tracer, *args, **kwargs):
-    # type: (...) -> Span
+def trace(weakdict: WeakValueDictionary, tracer: Tracer, *args, **kwargs) -> Span:
     """Return a span created from ``tracer`` and add it to the given weak
     dictionary.