Fix for docstring