diff --git a/timm/models/davit.py b/timm/models/davit.py
index 273ebcd0..58931cfd 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -341,7 +341,7 @@ class SpatialBlock(nn.Module):
         x = self.cpe[1](x, size)
         if self.ffn:
             x = x + self.drop_path(self.mlp(self.norm2(x)))
-        return x : Tensor, size
+        return x, size