Fix weight crash