diff --git a/src/compute/models/k_means.rs b/src/compute/models/k_means.rs
index 6dcd155..7fea599 100644
--- a/src/compute/models/k_means.rs
+++ b/src/compute/models/k_means.rs
@@ -256,11 +256,8 @@ mod tests {
         let mut sorted_labels = labels.clone();
         sorted_labels.sort_unstable();
         sorted_labels.dedup();
-        assert_eq!(
-            sorted_labels.len(),
-            k,
-            "Labels should all be unique when k==m"
-        );
+        // Labels should all be unique when k==m
+        assert_eq!(sorted_labels.len(), k);
     }
 
     #[test]