small cleanup in code review
small cleanup in code review