Fix documentation

Fix documentation