Fix doc comment

Closes #564