diff --git a/mp_rand_source.c b/mp_rand_source.c
index e9e87694..85ded746 100644
--- a/mp_rand_source.c
+++ b/mp_rand_source.c
@@ -1,5 +1,7 @@
 #include "tommath_private.h"
-#ifdef MP_RAND_C
+/* MP_RAND_C relates to mp_rand.c, MP_PRIME_RAND_C relates tp mp_prime_rand.c
+   Both use s_mp_rand_source */
+#if defined(MP_RAND_C) || defined(MP_PRIME_RAND_C)
 /* LibTomMath, multiple-precision integer library -- Tom St Denis */
 /* SPDX-License-Identifier: Unlicense */