Update docs to fix bug