Minor code review update