Fix trait type addition bug