diff --git a/src/kbmod/run_search.py b/src/kbmod/run_search.py
index 642c452a..608c725c 100644
--- a/src/kbmod/run_search.py
+++ b/src/kbmod/run_search.py
@@ -281,4 +281,4 @@ def run_search_from_work_unit(self, work):
         trj_generator = create_trajectory_generator(work.config, work_unit=work)
 
         # Run the search.
-        return self.run_search(work.config, work.im_stack)
+        return self.run_search(work.config, work.im_stack, trj_generator=trj_generator)