diff --git a/compiler_opt/rl/corpus.py b/compiler_opt/rl/corpus.py
index c351695e..3285611c 100644
--- a/compiler_opt/rl/corpus.py
+++ b/compiler_opt/rl/corpus.py
@@ -386,7 +386,7 @@ def reset(self):
   def sample(self, k: int, sort: bool = False) -> List[ModuleSpec]:
     """Samples `k` module_specs, optionally sorting by size descending.
 
-    Use load_corpus_element to get LoadedModuleSpecs - this allows the user
+    Use load_module_spec to get LoadedModuleSpecs - this allows the user to
     decide how the loading should happen (e.g. may want to use a threadpool)
     """
     # Note: sampler is intentionally defaulted to a mutable object, as the