Small code cleanup.