diff --git a/tests/actions.spec.ts b/tests/actions.spec.ts
index 95335481..bfa6adeb 100644
--- a/tests/actions.spec.ts
+++ b/tests/actions.spec.ts
@@ -26,7 +26,7 @@ test.describe('Check actions', () => {
 
     // Verify downloaded file is different for different diagrams
     await editPage.toggleSampleDiagrams();
-    await editPage.loadSampleDiagram('ER');
+    await editPage.loadSampleDiagram('Entity Relationship');
 
     const secondPngSize = await editPage.checkAndDownloadPNG(81_000);
     const secondSvgSize = await editPage.downloadSVG(11_000);
diff --git a/tests/docMap.spec.ts b/tests/docMap.spec.ts
index 649ac56a..e579a668 100644
--- a/tests/docMap.spec.ts
+++ b/tests/docMap.spec.ts
@@ -12,7 +12,7 @@ test.describe('Editor docs tests', () => {
   test('Test to see if the correct URL loads when changing from one diagram to other', async ({
     editPage
   }) => {
-    await editPage.loadSampleDiagram('Flow');
+    await editPage.loadSampleDiagram('Flowchart');
     await editPage.checkDocURL(/syntax\/flowchart\.html/);
     await editPage.setEditorMode('Config');
     await editPage.checkDocURL(/syntax\/flowchart\.html#configuration/);