code review change
minor fix