diff --git a/scripts/codegen/update_codegen_stamp.py b/scripts/codegen/update_codegen_stamp.py
index 290aee8232..81cbd60a97 100644
--- a/scripts/codegen/update_codegen_stamp.py
+++ b/scripts/codegen/update_codegen_stamp.py
@@ -29,7 +29,7 @@ def compute_combined_hash(input_files: list[str]) -> str:
     for filepath in input_files:
         if not Path(filepath).exists():
             print(f"Error: input file not found: {filepath}", file=sys.stderr)
-            sys.exit(1)
+            raise FileNotFoundError(f"Input file not found: {filepath}")
         file_hash = hashlib.sha256(Path(filepath).read_bytes()).hexdigest()
         parts.append(file_hash)