Tiny code cleanup
Tiny code cleanup