Simple code cleanup.