Minor code review fix.