diff --git a/frost-ed448/src/tests/batch.rs b/frost-ed448/src/tests/batch.rs
index 77fb83e..a4d911a 100644
--- a/frost-ed448/src/tests/batch.rs
+++ b/frost-ed448/src/tests/batch.rs
@@ -2,18 +2,14 @@ use rand::thread_rng;
 
 use crate::*;
 
-// TODO: make batching work for larger scalars
-// #[test]
-#[allow(unused)]
+#[test]
 fn check_batch_verify() {
     let rng = thread_rng();
 
     frost_core::tests::batch::batch_verify::<Ed448Shake256, _>(rng);
 }
 
-// TODO: make batching work for larger scalars
-// #[test]
-#[allow(unused)]
+#[test]
 fn check_bad_batch_verify() {
     let rng = thread_rng();
 
diff --git a/frost-ed448/src/tests/proptests.rs b/frost-ed448/src/tests/proptests.rs
index b01fd0e..a3b0fb2 100644
--- a/frost-ed448/src/tests/proptests.rs
+++ b/frost-ed448/src/tests/proptests.rs
@@ -1,11 +1,10 @@
+use crate::*;
 use frost_core::tests::proptests::{tweak_strategy, SignatureCase};
 use proptest::prelude::*;
 
 use rand_chacha::ChaChaRng;
 use rand_core::SeedableRng;
 
-use crate::*;
-
 proptest! {
 
     #[test]
diff --git a/gencode/src/main.rs b/gencode/src/main.rs
index ac5f0d5..55f41bd 100644
--- a/gencode/src/main.rs
+++ b/gencode/src/main.rs
@@ -282,6 +282,10 @@ fn main() -> ExitCode {
             "dkg.md",
             "src/keys/dkg.rs",
             "src/keys/repairable.rs",
+            "src/tests/batch.rs",
+            "src/tests/coefficient_commitment.rs",
+            "src/tests/proptests.rs",
+            "src/tests/vss_commitment.rs",
         ] {
             replaced |= copy_and_replace(
                 format!("{original_folder}/{filename}").as_str(),