Tiny tiny code cleanup