fix idmap indices

tskit-dev · Jun 11, 2024 · 46c1325 · 46c1325
1 parent e7e1e50
commit 46c1325
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test_fixtures.rs b/src/test_fixtures.rs
@@ -487,8 +487,8 @@ pub mod simulation {
             let pop_1 = tables.add_population().unwrap();
             let pop_2 = tables.add_population().unwrap();
             // get new ids after simplifcation
-            let child1 = id_map[0];
-            let child2 = id_map[1];
+            let child1 = id_map[child1.to_usize().unwrap()];
+            let child2 = id_map[child2.to_usize().unwrap()];
             tables
                 .add_migration((left, right), child1, (pop_anc, pop_1), t0 + 1.0)
                 .unwrap();