small code review change