diff --git a/crates/omnigraph/tests/staged_writes.rs b/crates/omnigraph/tests/staged_writes.rs
index 06fffc2..1d798a6 100644
--- a/crates/omnigraph/tests/staged_writes.rs
+++ b/crates/omnigraph/tests/staged_writes.rs
@@ -17,7 +17,7 @@
 //!    primitive's behavior so a future change either (a) preserves it or
 //!    (b) consciously fixes it (and updates this test).
 
-use arrow_array::{Int32Array, RecordBatch, StringArray};
+use arrow_array::{Array, Int32Array, RecordBatch, StringArray};
 use arrow_schema::{DataType, Field, Schema};
 use lance::dataset::{WhenMatched, WhenNotMatched};
 use omnigraph::table_store::TableStore;