diff --git a/node/actors/bft/src/leader/mod.rs b/node/actors/bft/src/leader/mod.rs
index 1ed65346..dfeff178 100644
--- a/node/actors/bft/src/leader/mod.rs
+++ b/node/actors/bft/src/leader/mod.rs
@@ -5,11 +5,12 @@
 mod replica_commit;
 mod replica_prepare;
 mod state_machine;
+
+pub(crate) use state_machine::StateMachine;
+
 #[cfg(test)]
 mod tests;
-
-#[allow(unused_imports)]
+#[cfg(test)]
 pub(crate) use replica_commit::Error as ReplicaCommitError;
-#[allow(unused_imports)]
+#[cfg(test)]
 pub(crate) use replica_prepare::Error as ReplicaPrepareError;
-pub(crate) use state_machine::StateMachine;
diff --git a/node/actors/bft/src/replica/mod.rs b/node/actors/bft/src/replica/mod.rs
index 822a4b48..40df624f 100644
--- a/node/actors/bft/src/replica/mod.rs
+++ b/node/actors/bft/src/replica/mod.rs
@@ -7,12 +7,13 @@ mod leader_commit;
 mod leader_prepare;
 mod new_view;
 mod state_machine;
-#[cfg(test)]
-mod tests;
 mod timer;
 
-#[allow(unused_imports)]
+pub(crate) use state_machine::StateMachine;
+
+#[cfg(test)]
+mod tests;
+#[cfg(test)]
 pub(crate) use leader_commit::Error as LeaderCommitError;
-#[allow(unused_imports)]
+#[cfg(test)]
 pub(crate) use leader_prepare::Error as LeaderPrepareError;
-pub(crate) use state_machine::StateMachine;