Add fix for traitsids