Tiny code cleanup