diff --git a/packages/core/src/tests/utility/mockOpenAI.ts b/packages/core/src/tests/utility/mockOpenAI.ts
index ccbadcf71a537a89fe64466c3a5286aa366660ac..c80617862d3a80d1cc6cb6c1a28f2b69500ded3e 100644
--- a/packages/core/src/tests/utility/mockOpenAI.ts
+++ b/packages/core/src/tests/utility/mockOpenAI.ts
@@ -62,12 +62,12 @@ export function mockLlmGeneration({
 }
 
 export function mockEmbeddingModel(embedModel: OpenAIEmbedding) {
-  jest.spyOn(embedModel, "aGetTextEmbedding").mockImplementation(async (x) => {
+  jest.spyOn(embedModel, "getTextEmbedding").mockImplementation(async (x) => {
     return new Promise((resolve) => {
       resolve([1, 0, 0, 0, 0, 0]);
     });
   });
-  jest.spyOn(embedModel, "aGetQueryEmbedding").mockImplementation(async (x) => {
+  jest.spyOn(embedModel, "getQueryEmbedding").mockImplementation(async (x) => {
     return new Promise((resolve) => {
       resolve([0, 1, 0, 0, 0, 0]);
     });