Fix documentation