fix test and add docstring