diff --git a/timm/models/mambaout.py b/timm/models/mambaout.py
index a33554a9c..c2f2f07b4 100644
--- a/timm/models/mambaout.py
+++ b/timm/models/mambaout.py
@@ -384,7 +384,7 @@ def __init__(
                 drop_rate=drop_rate,
             )
         self.num_features = prev_dim
-        self.hidden_size = self.head.num_features
+        self.head_hidden_size = self.head.num_features
 
         self.apply(self._init_weights)