[DOC] Fix example code syntax in code comment