A small code cleanup.