update docs and fix test