docstring + logic 'fix'