fix misplaced docstrings

Found using the eastwood lint tool.