diff --git a/timm/models/davit.py b/timm/models/davit.py
index 97f53fb6..35b6e00a 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -23,7 +23,7 @@ from typing import Any, Dict, Iterable, Iterator, Mapping, Optional, overload, T
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-import torch.Tensor as Tensor
+from torch import Tensor
 from .helpers import build_model_with_cfg
 from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 from .layers import DropPath, to_2tuple, trunc_normal_, SelectAdaptivePool2d, ClassifierHead, Mlp