Small code review