diff --git a/e2e/src/tests/replay.test.ts b/e2e/src/tests/replay.test.ts
index 24a8e8d740..4fa6518bec 100644
--- a/e2e/src/tests/replay.test.ts
+++ b/e2e/src/tests/replay.test.ts
@@ -875,7 +875,7 @@ test(
         Import.oldVersionAppFolderChatName,
       );
 
-      const newModels = [ModelIds.AWS_TITAN, ModelIds.GPT_4];
+      const newModels = [ModelIds.BISON_001, ModelIds.GPT_4];
       for (let i = 1; i <= newModels.length; i++) {
         await chatHeader.openConversationSettings.click();
         await talkToSelector.selectModel(
@@ -925,7 +925,7 @@ test(
         ...newMessages,
       ]);
       for (let i = 0; i < requests.length; i++) {
-        const modelId = i === 1 ? ModelIds.AWS_TITAN : ModelIds.GPT_4;
+        const modelId = i === 1 ? ModelIds.BISON_001 : ModelIds.GPT_4;
         expect
           .soft(requests[i].modelId, ExpectedMessages.chatRequestModelIsValid)
           .toBe(modelId);
diff --git a/e2e/src/tests/workWithModels.test.ts b/e2e/src/tests/workWithModels.test.ts
index 52a578501e..f8b90c8923 100644
--- a/e2e/src/tests/workWithModels.test.ts
+++ b/e2e/src/tests/workWithModels.test.ts
@@ -258,7 +258,8 @@ test('System prompt is applied in Model', async ({
   });
 });
 
-test('Stop generating for models like GPT (1 symbol = 1 token)', async ({
+//TODO: enable test when response is returned by chunks
+test.skip('Stop generating for models like GPT (1 symbol = 1 token)', async ({
   dialHomePage,
   chat,
   setTestIds,
@@ -323,7 +324,8 @@ test('Stop generating for models like GPT (1 symbol = 1 token)', async ({
   });
 });
 
-test('Send button in new message is available for Model if previous response is partly received when Stop generating was used', async ({
+//TODO: enable test when response is returned by chunks
+test.skip('Send button in new message is available for Model if previous response is partly received when Stop generating was used', async ({
   dialHomePage,
   chat,
   setTestIds,