diff --git a/mlkem/kem.h b/mlkem/kem.h
index 6d84a75d0..534e3783c 100644
--- a/mlkem/kem.h
+++ b/mlkem/kem.h
@@ -14,14 +14,6 @@
 #define CRYPTO_CIPHERTEXTBYTES MLKEM_CIPHERTEXTBYTES
 #define CRYPTO_BYTES MLKEM_SSBYTES
 
-#if (MLKEM_K == 2)
-#define CRYPTO_ALGNAME "Kyber512"
-#elif (MLKEM_K == 3)
-#define CRYPTO_ALGNAME "Kyber768"
-#elif (MLKEM_K == 4)
-#define CRYPTO_ALGNAME "Kyber1024"
-#endif
-
 #define crypto_kem_keypair_derand MLKEM_NAMESPACE(keypair_derand)
 /*************************************************
  * Name:        crypto_kem_keypair_derand
diff --git a/test/gen_NISTKAT.c b/test/gen_NISTKAT.c
index db7b571f6..91cfc0056 100644
--- a/test/gen_NISTKAT.c
+++ b/test/gen_NISTKAT.c
@@ -10,6 +10,14 @@
 #include "nistrng.h"
 #include "randombytes.h"
 
+#if (MLKEM_K == 2)
+#define CRYPTO_ALGNAME "Kyber512"
+#elif (MLKEM_K == 3)
+#define CRYPTO_ALGNAME "Kyber768"
+#elif (MLKEM_K == 4)
+#define CRYPTO_ALGNAME "Kyber1024"
+#endif
+
 static void fprintBstr(FILE *fp, const char *S, const uint8_t *A, size_t L)
 {
   size_t i;