Fix incorrect trait names