diff --git a/crates/lance-context/src/unified_generic.rs b/crates/lance-context/src/unified_generic.rs
index 86119d8..7a090a7 100644
--- a/crates/lance-context/src/unified_generic.rs
+++ b/crates/lance-context/src/unified_generic.rs
@@ -1,12 +1,13 @@
 use std::collections::HashMap;
 
 use lance_context_api::{
-    AddRowsResponse, ContextError, ContextResult, CreateGenericStoreRequest, GenericStoreApi,
-    SchemaSpec,
+    AddRowsResponse, ContextError, ContextResult, GenericStoreApi, SchemaSpec,
 };
 use lance_context_core::{GenericStore as LocalStore, GenericStoreOptions};
 use serde_json::{Map, Value};
 
+#[cfg(feature = "remote")]
+use lance_context_api::CreateGenericStoreRequest;
 #[cfg(feature = "remote")]
 use lance_context_client::RemoteGenericStore;