diff --git a/src/dataset/generator.rs b/src/dataset/generator.rs
index 4d454af..4367308 100644
--- a/src/dataset/generator.rs
+++ b/src/dataset/generator.rs
@@ -164,8 +164,7 @@ mod tests {
 
     #[test]
     fn test_make_moons() {
-        let dataset = make_moons(100, 0.05);
-        println!("{:?}", dataset.data);
+        let dataset = make_moons(10, 0.05);
         assert_eq!(
             dataset.data.len(),
             dataset.num_features * dataset.num_samples