Put docstring in the correct place

Fixes #50