Fix model width extraction