Update weight parser to better handle short exports.