Tweaks to trait