diff --git a/scdl/scdl.py b/scdl/scdl.py
index 2c4cf9c..4fc8fe8 100644
--- a/scdl/scdl.py
+++ b/scdl/scdl.py
@@ -285,7 +285,7 @@ def __enter__(self):
 
     def __exit__(
         self,
-        exc_type: Optional[type[BaseException]],
+        exc_type: Optional[Type[BaseException]],
         exc_value: Optional[BaseException],
         traceback: Optional[TracebackType],
     ) -> None: