diff --git a/src/semra/pipeline.py b/src/semra/pipeline.py
index 6bcfe0c..6a26afd 100644
--- a/src/semra/pipeline.py
+++ b/src/semra/pipeline.py
@@ -279,10 +279,10 @@ def get_raw_mappings(configuration: Configuration) -> t.List[Mapping]:
 def process(
     mappings: t.List[Mapping],
     upgrade_prefixes=None,
-    remove_prefix_set: t.Optional[t.Collection[str]]=None,
-    keep_prefix_set: t.Optional[t.Collection[str]]=None,
-    post_remove_prefixes: t.Optional[t.Collection[str]]=None,
-    post_keep_prefixes: t.Optional[t.Collection[str]]=None,
+    remove_prefix_set: t.Optional[t.Collection[str]] = None,
+    keep_prefix_set: t.Optional[t.Collection[str]] = None,
+    post_remove_prefixes: t.Optional[t.Collection[str]] = None,
+    post_keep_prefixes: t.Optional[t.Collection[str]] = None,
     *,
     remove_imprecise: bool = True,
 ) -> t.List[Mapping]:
diff --git a/src/semra/sources/wikidata.py b/src/semra/sources/wikidata.py
index 42930b4..9f18994 100644
--- a/src/semra/sources/wikidata.py
+++ b/src/semra/sources/wikidata.py
@@ -7,7 +7,10 @@
 from semra.rules import EXACT_MATCH, UNSPECIFIED_MAPPING
 from semra.struct import Mapping, MappingSet, SimpleEvidence
 
-__all__ = ["get_wikidata_mappings"]
+__all__ = [
+    "get_wikidata_mappings",
+    "get_wikidata_mappings_by_prefix",
+]
 
 
 def get_wikidata_mappings(prop: str, predicate: t.Optional[Reference] = None) -> t.List[Mapping]: