Fix wrong data type in docstring