diff --git a/ipa-core/src/utils/power_of_two.rs b/ipa-core/src/utils/power_of_two.rs
index abce8055e..a84455c92 100644
--- a/ipa-core/src/utils/power_of_two.rs
+++ b/ipa-core/src/utils/power_of_two.rs
@@ -68,7 +68,7 @@ impl NonZeroU32PowerOfTwo {
     }
 }
 
-#[cfg(test)]
+#[cfg(all(test, unit_test))]
 mod tests {
     use super::{ConvertError, NonZeroU32PowerOfTwo};