Fixed the unit heat not being updated after firing weapons