diff --git a/slide2vec/models/models.py b/slide2vec/models/models.py
index 0f835c8..cc93daf 100644
--- a/slide2vec/models/models.py
+++ b/slide2vec/models/models.py
@@ -711,7 +711,7 @@ def forward(self, x):
 
 class Midnight12k(FeatureExtractor):
     def __init__(self):
-        self.features_dim = 1536
+        self.features_dim = 3072
         super(Midnight12k, self).__init__()
 
     def build_encoder(self):
@@ -749,7 +749,7 @@ def build_encoder(self):
 
     def get_transforms(self):
         return AutoImageProcessor.from_pretrained(
-            "histai/hibou-L", trust_remote_code=True
+            f"histai/{self.arch}", trust_remote_code=True
         )
 
     def forward(self, x):