diff --git a/prompts/dafny_invariants_cot/001.txt b/prompts/dafny_invariants_cot/steps/001.txt
similarity index 100%
rename from prompts/dafny_invariants_cot/001.txt
rename to prompts/dafny_invariants_cot/steps/001.txt
diff --git a/verified_cogen/runners/step_by_step.py b/verified_cogen/runners/step_by_step.py
index 95cdf07..eaadc07 100644
--- a/verified_cogen/runners/step_by_step.py
+++ b/verified_cogen/runners/step_by_step.py
@@ -17,7 +17,7 @@ def preprocess(self, prg: str, mode: Mode) -> str:
 
     def rewrite(self, prg: str) -> str:
         steps: list[Step] = []
-        for step in sorted(pathlib.Path("prompts/step_by_step").glob("*.txt")):
+        for step in sorted((pathlib.Path(self.llm.prompt_dir) / "steps").glob("*.txt")):
             steps.append(Step(step.read_text()))
         for step in steps:
             for substep in step.substeps: