diff --git a/timm/models/davit.py b/timm/models/davit.py
index 5d2a8530..d7308e38 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -35,9 +35,8 @@ from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 __all__ = ['DaViT']
 
 class SequentialWithSize(nn.Sequential):
-
-    def __init__(*args, **kwargs):
-        super(MySequential, self).__init__(*args, **kwargs)
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
 
     def forward(self, x: Tensor, size: Tuple[int, int]):
         for module in self._modules.values():