[Cleanup] slight code cleanup.