Don't like the duplication of this logic. Would splitting it out in to a method make it DRY?