Update Trait picker when changing trait set

Fixes #196