[MINOR] Small cleanup