fixed docstring and test