Fix docs bug