diff --git a/tests/data_files/Makefile b/tests/data_files/Makefile
index 01d2379..e8917f1 100644
--- a/tests/data_files/Makefile
+++ b/tests/data_files/Makefile
@@ -2157,6 +2157,9 @@
 				--string TEST_CLI_KEY_RSA_PEM=cli-rsa.key \
 				--binary TEST_CLI_KEY_RSA_DER=cli-rsa.key.der
 
+../src/test_keys.h: ../scripts/generate_test_keys.py
+	../scripts/generate_test_keys.py --output $@
+
 ################################################################
 #### Diffie-Hellman parameters
 ################################################################
diff --git a/tests/scripts/generate_test_keys.py b/tests/scripts/generate_test_keys.py
index 0a67a78..9920933 100755
--- a/tests/scripts/generate_test_keys.py
+++ b/tests/scripts/generate_test_keys.py
@@ -9,6 +9,7 @@
 import os
 from typing import Iterator
 import re
+import argparse
 import scripts_path # pylint: disable=unused-import
 from mbedtls_dev.asymmetric_key_data import ASYMMETRIC_KEY_DATA
 
@@ -90,11 +91,17 @@
     yield "      {0}, sizeof({0}) }},".format(pub_array_name)
 
 def main() -> None:
-    # Remove output file if already existing.
-    if os.path.exists(OUTPUT_HEADER_FILE):
-        os.remove(OUTPUT_HEADER_FILE)
+    argparser = argparse.ArgumentParser()
+    argparser.add_argument("--output", required=True, help="Output file")
+    args = argparser.parse_args()
 
-    output_file = open(OUTPUT_HEADER_FILE, 'at')
+    output_file = args.output
+    # Remove output file if already existing.
+    if os.path.exists(output_file):
+        print("Warning: {} already existing, it will be overwritten.", output_file)
+        os.remove(output_file)
+
+    output_file = open(output_file, 'at')
     output_file.write(
         "/*********************************************************************************\n" +
         " * This file was automatically generated from tests/scripts/generate_test_keys.py.\n" +