Fixing trait adding search