Update doc in comment