Minor code polish.