Minor fixes on the doc-test