Update docs, fix test