diff --git a/tests/vector_stores/test_databricks.py b/tests/vector_stores/test_databricks.py
index 50f074b30..93c0324d5 100644
--- a/tests/vector_stores/test_databricks.py
+++ b/tests/vector_stores/test_databricks.py
@@ -5,7 +5,6 @@ import pytest
 
 pytest.importorskip("databricks", reason="databricks-sdk package not installed")
 
-from databricks.sdk.service.sql import StatementParameterListItem
 from databricks.sdk.service.vectorsearch import VectorIndexType, QueryVectorIndexResponse, ResultManifest, ResultData, ColumnInfo
 from mem0.vector_stores.databricks import Databricks