diff --git a/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py b/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py
index 4be04455392..36f4c7d3a21 100644
--- a/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py
+++ b/libs/partners/pinecone/tests/unit_tests/test_vectorstores.py
@@ -1,6 +1,6 @@
 from unittest.mock import Mock
 
-from langchain_pinecone.vectorstores import PineconeVectorStore
+from langchain_pinecone.vectorstores import Pinecone, PineconeVectorStore
 
 
 def test_initialization() -> None:
@@ -9,7 +9,7 @@ def test_initialization() -> None:
     index = Mock()
     embedding = Mock()
     text_key = "xyz"
-    PineconeVectorStore(index, embedding, text_key)
+    Pinecone(index, embedding, text_key)
 
 
 def test_id_prefix() -> None: