diff --git a/omnidata_tools/torch/modules/midas/vit.py b/omnidata_tools/torch/modules/midas/vit.py
index 80ba7ad..fe0e4a7 100644
--- a/omnidata_tools/torch/modules/midas/vit.py
+++ b/omnidata_tools/torch/modules/midas/vit.py
@@ -480,7 +480,7 @@ def _make_vit_b_rn50_backbone(
 def _make_pretrained_vitb_rn50_384(
     pretrained, use_readout="ignore", hooks=None, use_vit_only=False
 ):
-    model = timm.create_model("vit_base_resnet50_384", pretrained=pretrained)
+    model = timm.create_model("vit_base_r50_s16_384", pretrained=pretrained)
 
     hooks = [0, 1, 8, 11] if hooks == None else hooks
     return _make_vit_b_rn50_backbone(