Super minor code cleanup