diff --git a/src/model.rs b/src/model.rs
index 78718a4..746edb4 100644
--- a/src/model.rs
+++ b/src/model.rs
@@ -64,11 +64,14 @@ impl Model {
         for i in 0..n_populations {
             attraction_table.push(Vec::with_capacity(n_populations));
             for j in 0..n_populations {
-                attraction_table[i].push(if i == j {
-                    attraction_distr.sample(&mut rng)
-                } else {
-                    repulsion_distr.sample(&mut rng)
-                });
+                attraction_table[i].push(
+                    if i == j {
+                        &attraction_distr
+                    } else {
+                        &repulsion_distr
+                    }
+                    .sample(&mut rng),
+                );
             }
         }