[minor] Make code more readable