update documentation + fix test