Code review fix.