docs + fix test