Fix docs to make it work