fix previous commit

... testing does help sometimes ...
a little bit better
better docstring