Update to use as a trait.