Chenge docstring to the correct test class