diff --git a/crates/oa-desktop/src/connection_allocation_tests.rs b/crates/oa-desktop/src/connection_allocation_tests.rs
index de132fa8..2f8c93e6 100644
--- a/crates/oa-desktop/src/connection_allocation_tests.rs
+++ b/crates/oa-desktop/src/connection_allocation_tests.rs
@@ -21,7 +21,10 @@ fn allocation_fixture(layer: Layer, projection: ProjectionKind) -> (App, Id, Id,
     project.place(reference_id, performer, Position { x: 80., y: 100. });
     project.place(reference_id, work, Position { x: 500., y: 300. });
     let mut app = App::with_project(project);
-    let _ = app.update(Message::Diagram(diagram));
+    let _ = app.update(Message::Diagram(crate::app::Choice {
+        id: diagram,
+        label: app.project().diagrams[&diagram].name.clone(),
+    }));
     (app, performer, work, diagram)
 }
 
@@ -228,7 +231,10 @@ fn allocation_candidates_reject_nonleaf_occupied_cross_layer_port_and_hidden_wor
         .unwrap();
         project.elements.insert(port_id, port);
         let mut app = App::with_project(project);
-        let _ = app.update(Message::Diagram(diagram));
+        let _ = app.update(Message::Diagram(crate::app::Choice {
+            id: diagram,
+            label: app.project().diagrams[&diagram].name.clone(),
+        }));
         let before = app.document.bundle().clone();
         let original = app.project().clone();
         let revision = app.document.revision();
@@ -286,7 +292,10 @@ fn allocation_creation_preserves_saved_edge_filters() {
         .relation_kinds = oa_core::view_options::RelationKinds::Only(vec![]);
     let diagrams = project.diagrams.clone();
     let mut app = App::with_project(project);
-    let _ = app.update(Message::Diagram(diagram));
+    let _ = app.update(Message::Diagram(crate::app::Choice {
+        id: diagram,
+        label: app.project().diagrams[&diagram].name.clone(),
+    }));
     let context = start_allocation(&mut app, performer);
     send(&mut app, Event::Endpoint(context, work));
     assert_eq!(
@@ -320,7 +329,10 @@ fn allocation_mixed_view_keeps_captured_canvas_for_opposite_layer_pairs() {
             project.place(diagram, performer, Position { x: 40., y: 350. });
             project.place(diagram, work, Position { x: 420., y: 350. });
             let mut app = App::with_project(project);
-            let _ = app.update(Message::Diagram(diagram));
+            let _ = app.update(Message::Diagram(crate::app::Choice {
+                id: diagram,
+                label: app.project().diagrams[&diagram].name.clone(),
+            }));
             let _ = app.update(Message::Pan(iced::Vector::new(73., 29.)));
             app.layer = app_layer;
             let camera = (app.pan, app.zoom);
@@ -402,7 +414,10 @@ fn allocation_cancel_navigation_stale_context_and_pending_edits_preserve_model()
     let _ = app.update(Message::Page(Page::Catalog));
     assert!(app.connection.is_none());
     let _ = app.update(Message::Page(Page::Diagram));
-    let _ = app.update(Message::Diagram(diagram));
+    let _ = app.update(Message::Diagram(crate::app::Choice {
+        id: diagram,
+        label: app.project().diagrams[&diagram].name.clone(),
+    }));
     start_allocation(&mut app, performer);
     send(&mut app, Event::Choose(None));
     assert!(app.connection.is_none());