Fixed a small bug caused by docstrings