Make it clear method returns a set. Also use a feature for a better test failure message.