Add spec for set helper method