Fixing trait correctly