Slight code cleanup.
Slight code cleanup.