diff --git a/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py b/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py
index c4f4568f..77afc9e1 100644
--- a/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py
+++ b/libs/aws/tests/integration_tests/vectorstores/test_documentdb.py
@@ -11,7 +11,7 @@
 from motor.core import AgnosticCollection
 from pymongo.collection import Collection
 
-from langchain_aws.vectorstores.documentdb_new import (
+from langchain_aws.vectorstores.documentdb import (
     DocumentDBSimilarityType,
     DocumentDBVectorSearch,
 )