diff --git a/extension/src/api.rs b/extension/src/api.rs
index 1ea2c5d..d2efa9c 100644
--- a/extension/src/api.rs
+++ b/extension/src/api.rs
@@ -105,7 +105,7 @@ fn init_rag(
         columns,
         unique_record_id,
         None,
-        index_dist_type,
+        index_dist_type.into(),
         &transformer_model,
         table_method.into(),
         schedule,
diff --git a/extension/src/types.rs b/extension/src/types.rs
index e28df65..397d373 100644
--- a/extension/src/types.rs
+++ b/extension/src/types.rs
@@ -1,5 +1,7 @@
 use pgrx::*;
-use vectorize_core::types::{IndexDist as CoreIndexDist, TableMethod as CoreTableMethod};
+use vectorize_core::types::{
+    IndexDist as CoreIndexDist, SimilarityAlg as CoreSimilarityAlg, TableMethod as CoreTableMethod,
+};
 
 use serde::{Deserialize, Serialize};
 pub const VECTORIZE_SCHEMA: &str = "vectorize";