diff --git a/tests/data_files/Makefile b/tests/data_files/Makefile
index 01d2379d1e..e8917f170f 100644
--- a/tests/data_files/Makefile
+++ b/tests/data_files/Makefile
@@ -2157,6 +2157,9 @@ TEST_CERTS_H_INPUT_FILES=test-ca2.crt \
 				--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 0a67a784cb..9920933692 100755
--- a/tests/scripts/generate_test_keys.py
+++ b/tests/scripts/generate_test_keys.py
@@ -9,6 +9,7 @@ generating the required key at run time. This helps speeding up testing."""
 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 @@ def get_look_up_table_entry(key_type: str, group_id_or_keybits: str,
     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" +