diff --git a/demo/scripts/playwright-docker.sh b/demo/scripts/playwright-docker.sh
index 287b5d74e..312d436ad 100755
--- a/demo/scripts/playwright-docker.sh
+++ b/demo/scripts/playwright-docker.sh
@@ -66,8 +66,14 @@ fi
 
 if [[ "$1" = "update" ]]; then
   echo "Running playwright tests (update)"
+  EXTRA_ARGS=""
+  if [[ $# -ge 2 ]]; then
+    for arg in "${@:2}"; do
+      EXTRA_ARGS="$EXTRA_ARGS $(printf '%q' "$arg")"
+    done
+  fi
   run_command 'COREPACK_INTEGRITY_KEYS=0 corepack pnpm --filter '@gravity-ui/*' build'
-  run_command 'COREPACK_INTEGRITY_KEYS=0 corepack pnpm --filter '@markdown-editor/demo' run playwright:update'
+  run_command "cd demo && COREPACK_INTEGRITY_KEYS=0 corepack pnpm exec playwright test --config=tests/playwright/playwright.config.ts -u $EXTRA_ARGS"
   exit 0
 fi
 
diff --git a/demo/src/stories/markdown/Markdown.stories.tsx b/demo/src/stories/markdown/Markdown.stories.tsx
index b86c08df2..8561c4662 100644
--- a/demo/src/stories/markdown/Markdown.stories.tsx
+++ b/demo/src/stories/markdown/Markdown.stories.tsx
@@ -61,7 +61,7 @@ export const PunctuationBoundaries: StoryObj<typeof component> = {
         initial: markup.punctuationBoundaries,
         initialEditor: 'markup',
         initialSplitModeEnabled: true,
-        splitModeOrientation: 'horizontal',
+        splitModeOrientation: 'vertical',
     },
 };
 
diff --git a/demo/tests/playwright/core/mount.tsx b/demo/tests/playwright/core/mount.tsx
index ce23dab41..07afa6119 100644
--- a/demo/tests/playwright/core/mount.tsx
+++ b/demo/tests/playwright/core/mount.tsx
@@ -35,6 +35,7 @@ export const mount: PlaywrightFixture<MountFixture> = async ({mount: baseMount},
                         `}
                     </style>
                 )}
+                {options?.styles && <style>{options.styles}</style>}
                 {component}
             </div>,
             options,
diff --git a/demo/tests/playwright/core/types.ts b/demo/tests/playwright/core/types.ts
index a45bfeb7f..eda406a16 100644
--- a/demo/tests/playwright/core/types.ts
+++ b/demo/tests/playwright/core/types.ts
@@ -24,6 +24,7 @@ interface ComponentFixtures {
             width?: number | string;
             rootStyle?: React.CSSProperties;
             hidePlaygroundBlocks?: boolean;
+            styles?: string;
         },
     ): Promise<MountResult>;
 }
diff --git a/demo/tests/visual-tests/MarkdownExtensions.visual.test.tsx b/demo/tests/visual-tests/MarkdownExtensions.visual.test.tsx
index 7830545b3..261e4be76 100644
--- a/demo/tests/visual-tests/MarkdownExtensions.visual.test.tsx
+++ b/demo/tests/visual-tests/MarkdownExtensions.visual.test.tsx
@@ -51,18 +51,12 @@ test.describe('Extensions, Markdown', () => {
         await mount(<MarkdownStories.DefinitionList />);
         await expectScreenshot();
     });
-    test('Punctuation boundaries', async ({mount, expectScreenshot, page}) => {
-        await page.setViewportSize({width: 1280, height: 1400});
+    test('Punctuation boundaries', async ({page, mount, expectScreenshot}) => {
         await mount(<MarkdownStories.PunctuationBoundaries />, {
-            rootStyle: {height: 'auto', width: 1200},
-        });
-        await page.addStyleTag({
-            content:
-                '.cm-editor { height: auto !important; } .cm-scroller { overflow: visible !important; }',
+            hidePlaygroundBlocks: true,
+            styles: '.cm-editor { height: auto !important; } .cm-scroller { overflow: visible !important; }',
         });
         await page.waitForTimeout(300);
-        await expectScreenshot({
-            component: page.locator('.playwright-wrapper-test'),
-        });
+        await expectScreenshot();
     });
 });
diff --git a/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-dark-chromium-linux.png b/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-dark-chromium-linux.png
index bf2e8d93e..6a5613380 100644
Binary files a/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-dark-chromium-linux.png and b/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-dark-chromium-linux.png differ
diff --git a/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-light-chromium-linux.png b/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-light-chromium-linux.png
index 83b445879..7c2e4b111 100644
Binary files a/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-light-chromium-linux.png and b/demo/tests/visual-tests/__snapshots__/MarkdownExtensions.visual.test.tsx-snapshots/Extensions-Markdown-Punctuation-boundaries-light-chromium-linux.png differ