Fix documentation in comment