diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_directory.py b/packages/jupyter-ai/jupyter_ai/tests/test_directory.py
index 0695c2e10..403c5c3d5 100644
--- a/packages/jupyter-ai/jupyter_ai/tests/test_directory.py
+++ b/packages/jupyter-ai/jupyter_ai/tests/test_directory.py
@@ -1,5 +1,5 @@
 """
-Test that the collect_files function only selects files that are
+Test that the `collect_filepaths` function only selects files that are
 1. Not in the the excluded directories and
 2. Are in the  valid file extensions list.
 """