Tiny little code cleanup