Fix doc-tests