diff --git a/tests/scripts/generate_psa_tests.py b/tests/scripts/generate_psa_tests.py
index 492810b..d346631 100755
--- a/tests/scripts/generate_psa_tests.py
+++ b/tests/scripts/generate_psa_tests.py
@@ -78,6 +78,11 @@
     """
     used = set()
     for expr in expressions:
+        # HKDF_EXTRACT and HKDF_EXPAND algs depend on HKDF
+        if "HKDF_EXTRACT" in expr:
+            expr = expr.replace("HKDF_EXTRACT", "HKDF")
+        if "HKDF_EXPAND" in expr:
+            expr = expr.replace("HKDF_EXPAND", "HKDF")
         used.update(re.findall(r'PSA_(?:ALG|ECC_FAMILY|KEY_TYPE)_\w+', expr))
     used.difference_update(SYMBOLS_WITHOUT_DEPENDENCY)
     return sorted(psa_want_symbol(name) for name in used)