Fix documentation on docs.rs

Fix documentation on docs.rs #2401