diff --git a/sdk/translation/ai-translation-document-rest/test/public/utils/recordedClient.ts b/sdk/translation/ai-translation-document-rest/test/public/utils/recordedClient.ts
index b599a215fa2e..4d220e1ab5ba 100644
--- a/sdk/translation/ai-translation-document-rest/test/public/utils/recordedClient.ts
+++ b/sdk/translation/ai-translation-document-rest/test/public/utils/recordedClient.ts
@@ -4,8 +4,7 @@
 import { Context } from "mocha";
 import { Recorder, env } from "@azure-tools/test-recorder";
 import { ClientOptions } from "@azure-rest/core-client";
-import { DocumentTranslationClient } from "../../../src";
-import createClient from "../../../src/documentTranslationClient";
+import { DocumentTranslationClient, default as createClient } from "../../../src";
 import { KeyCredential, TokenCredential } from "@azure/core-auth";
 import { createTestCredential } from "@azure-tools/test-credential";
 
diff --git a/sdk/translation/ai-translation-document-rest/test/public/utils/testHelper.ts b/sdk/translation/ai-translation-document-rest/test/public/utils/testHelper.ts
index 3caa6d157c96..54e3a157491c 100644
--- a/sdk/translation/ai-translation-document-rest/test/public/utils/testHelper.ts
+++ b/sdk/translation/ai-translation-document-rest/test/public/utils/testHelper.ts
@@ -10,7 +10,7 @@ import {
   StorageInputType,
   StorageSource,
   TargetInput,
-} from "../../../src/models";
+} from "../../../src";
 import { TestDocument, createTestDocument } from "./TestDocument";
 
 export function createSourceInput(