diff --git a/src/ragas/testset/generator.py b/src/ragas/testset/generator.py
index a3f682c59..de5c33630 100644
--- a/src/ragas/testset/generator.py
+++ b/src/ragas/testset/generator.py
@@ -97,6 +97,7 @@ def with_openai(
             generator_llm_model=generator_llm_model,
             embeddings_model=embeddings_model,
             critic_llm_model=critic_llm_model,
+            docstore=docstore,
             run_config=run_config,
         )
 
@@ -146,7 +147,7 @@ def _common_constructor(
         cls,
         chunk_size: int,
         generator_llm_model: LangchainLLMWrapper,
-        embeddings_model: LangchainLLMWrapper,
+        embeddings_model: BaseRagasEmbeddings,
         critic_llm_model: LangchainLLMWrapper,
         docstore: t.Optional[DocumentStore],
         run_config: t.Optional[RunConfig],