Fix trait bugs