diff --git a/sdk/program-memory/src/lib.rs b/sdk/program-memory/src/lib.rs
index 79d0d20946fcf7..376b1fc215d2dc 100644
--- a/sdk/program-memory/src/lib.rs
+++ b/sdk/program-memory/src/lib.rs
@@ -220,7 +220,7 @@ pub fn sol_memset(s: &mut [u8], c: u8, n: usize) {
     }
 }
 
-#[cfg(test)]
+#[cfg(all(feature = "num-traits", test))]
 mod tests {
     use super::*;