diff --git a/tfhe-csprng/src/generators/default.rs b/tfhe-csprng/src/generators/default.rs
index 1e3e8f27bf..4914b37efd 100644
--- a/tfhe-csprng/src/generators/default.rs
+++ b/tfhe-csprng/src/generators/default.rs
@@ -1,19 +1,9 @@
 #[cfg(all(target_arch = "x86_64", not(feature = "software-prng")))]
-use super::AesniRandomGenerator;
+pub type DefaultRandomGenerator = super::AesniRandomGenerator;
 #[cfg(all(target_arch = "aarch64", not(feature = "software-prng")))]
-use super::NeonAesRandomGenerator;
+pub type DefaultRandomGenerator = super::NeonAesRandomGenerator;
 #[cfg(any(
     feature = "software-prng",
     not(any(target_arch = "x86_64", target_arch = "aarch64"))
 ))]
-use super::SoftwareRandomGenerator;
-
-#[cfg(all(target_arch = "x86_64", not(feature = "software-prng")))]
-pub type DefaultRandomGenerator = AesniRandomGenerator;
-#[cfg(all(target_arch = "aarch64", not(feature = "software-prng")))]
-pub type DefaultRandomGenerator = NeonAesRandomGenerator;
-#[cfg(any(
-    feature = "software-prng",
-    not(any(target_arch = "x86_64", target_arch = "aarch64"))
-))]
-pub type DefaultRandomGenerator = SoftwareRandomGenerator;
+pub type DefaultRandomGenerator = super::SoftwareRandomGenerator;