diff --git a/iris-mpc-gpu/src/threshold_ring/protocol.rs b/iris-mpc-gpu/src/threshold_ring/protocol.rs
index 826b0598c..dbaae5bc6 100644
--- a/iris-mpc-gpu/src/threshold_ring/protocol.rs
+++ b/iris-mpc-gpu/src/threshold_ring/protocol.rs
@@ -2406,6 +2406,7 @@ impl Circuits {
 
     // same as compare_threshold_masked_many, just via the functions used in the
     // bucketing
+    // Just here for testing
     pub fn compare_threshold_masked_many_bucket_functions(
         &mut self,
         code_dots: &[ChunkShareView<u16>],
diff --git a/iris-mpc-gpu/tests/bucket_threshold.rs b/iris-mpc-gpu/tests/bucket_threshold.rs
index eb8d5bf76..b51348be2 100644
--- a/iris-mpc-gpu/tests/bucket_threshold.rs
+++ b/iris-mpc-gpu/tests/bucket_threshold.rs
@@ -1,5 +1,5 @@
 #[cfg(feature = "gpu_dependent")]
-mod threshold_test {
+mod bucket_threshold_test {
     use cudarc::{
         driver::{CudaDevice, CudaStream},
         nccl::Id,
@@ -230,7 +230,7 @@ mod threshold_test {
     }
 
     #[tokio::test(flavor = "multi_thread", worker_threads = 3)]
-    async fn test_threshold() -> eyre::Result<()> {
+    async fn test_bucket_threshold() -> eyre::Result<()> {
         install_tracing();
         env::set_var("NCCL_P2P_LEVEL", "LOC");
         env::set_var("NCCL_NET", "Socket");