Slight refactor for style.