Small refactor, nothing great - makes more sense when reading.