Correct docstring param type