Fixed a minor trait issue