diff --git a/sphinx_d2lang/__init__.py b/sphinx_d2lang/__init__.py
index 0ef71de..1d6fcd7 100644
--- a/sphinx_d2lang/__init__.py
+++ b/sphinx_d2lang/__init__.py
@@ -2,6 +2,8 @@ from sphinx.application import Sphinx
 
 from .d2lang import D2langDirective
 
+from typing import Any, Dict, List, Tuple
+
 def setup(application: Sphinx) -> Dict[str, Any]:
     """
     setup extension.