Fix documentation generator