diff --git a/tests/unit/codegen/cli/conftest.py b/tests/unit/codegen/cli/conftest.py
index db8cd312f..3263638ca 100644
--- a/tests/unit/codegen/cli/conftest.py
+++ b/tests/unit/codegen/cli/conftest.py
@@ -1,4 +1,5 @@
 import os
+import shutil
 import subprocess
 from pathlib import Path
 
@@ -30,4 +31,5 @@ def initialized_repo(sample_repository: Path, runner: CliRunner):
     runner.invoke(init_command)
     subprocess.run(["git", "add", "."], cwd=sample_repository, check=True)
     subprocess.run(["git", "commit", "-m", "Initialize codegen"], cwd=sample_repository, check=True)
-    return sample_repository
+    yield sample_repository
+    shutil.rmtree(sample_repository)