diff --git a/testing/downstream_testing/downstream_runner.py b/testing/downstream_testing/downstream_runner.py
index a3fd44109..2f2ab4d2a 100644
--- a/testing/downstream_testing/downstream_runner.py
+++ b/testing/downstream_testing/downstream_runner.py
@@ -15,7 +15,9 @@ from typing import Match
 from typing import TYPE_CHECKING
 
 import yaml
-from typing_extensions import TypedDict
+
+if TYPE_CHECKING:
+    from typing_extensions import TypedDict
 
 logging.basicConfig(
     format="%(levelname)s | %(module)s.%(funcName)s | %(message)s", level="INFO"