Fix documentation for recent changes