FIX: correct docstring