diff --git a/timm/models/davit.py b/timm/models/davit.py
index 41af0ebe..464b976f 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -510,7 +510,7 @@ class DaViT(nn.Module):
         
         
         
-        for patch_layer, blocks in itertools.izip(self.patch_embeds, self.main_blocks):
+        for patch_layer, blocks in zip(self.patch_embeds, self.main_blocks):
             features[-1], sizes[-1] = patch_layer(features[-1], sizes[-1])
             
             for layer in enumerate(blocks):