diff --git a/timm/models/davit.py b/timm/models/davit.py
index b11342b8..3f0d7107 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -14,7 +14,7 @@ DaViT model defs and weights adapted from https://github.com/dingmyu/davit, orig
 
 import itertools
 
-from typing import Any, Dict, Iterable, Iterator, Mapping, Optional, overload, Tuple, TypeVar, Union
+from typing import Any, Dict, Iterable, Iterator, Mapping, Optional, overload, Tuple, TypeVar, Union, List