[MINOR] Clean up