diff --git a/scripts/generate_tls_handshake_tests.py b/scripts/generate_tls_handshake_tests.py
index cc1325661..7fdb5528c 100755
--- a/scripts/generate_tls_handshake_tests.py
+++ b/scripts/generate_tls_handshake_tests.py
@@ -8,6 +8,7 @@ Generate miscellaneous TLS test cases relating to the handshake.
 # SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
 
 import argparse
+import os
 import sys
 from typing import Optional
 
@@ -149,7 +150,7 @@ def write_handshake_tests(out: typing_util.Writable) -> None:
     out.write(f"""\
 # Miscellaneous tests related to the TLS handshake layer.
 #
-# Automatically generated by {sys.argv[0]}. Do not edit!
+# Automatically generated by {os.path.basename(sys.argv[0])}. Do not edit!
 
 # Copyright The Mbed TLS Contributors
 # SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later