Doc fix

Closes #2