fix bug with trait, I think