Fix docs, test