diff --git a/eth2/proto_array_fork_choice/src/bin.rs b/eth2/proto_array_fork_choice/src/bin.rs
index ee93bd407f..dec53a4ed5 100644
--- a/eth2/proto_array_fork_choice/src/bin.rs
+++ b/eth2/proto_array_fork_choice/src/bin.rs
@@ -1,12 +1,4 @@
-#[allow(dead_code)]
-mod error;
-mod fork_choice_test_definition;
-mod proto_array;
-#[allow(dead_code)]
-mod proto_array_fork_choice;
-mod ssz_container;
-
-pub use fork_choice_test_definition::*;
+use proto_array_fork_choice::fork_choice_test_definition::*;
 use serde_yaml;
 use std::fs::File;
 
diff --git a/eth2/proto_array_fork_choice/src/lib.rs b/eth2/proto_array_fork_choice/src/lib.rs
index 85ac466f43..65134e3e66 100644
--- a/eth2/proto_array_fork_choice/src/lib.rs
+++ b/eth2/proto_array_fork_choice/src/lib.rs
@@ -1,4 +1,5 @@
 mod error;
+pub mod fork_choice_test_definition;
 mod proto_array;
 mod proto_array_fork_choice;
 mod ssz_container;