diff --git a/src/web/App.mobile-layout.test.tsx b/src/web/App.mobile-layout.test.tsx
index a80cada..f9f2e3f 100644
--- a/src/web/App.mobile-layout.test.tsx
+++ b/src/web/App.mobile-layout.test.tsx
@@ -7,6 +7,6 @@ describe('App mobile layout', () => {
     const source = readFileSync(resolve(process.cwd(), 'src/web/App.tsx'), 'utf8');
 
     expect(source).toContain('useIsMobile(768)');
-    expect(source).toContain("document.documentElement.setAttribute('data-layout'");
+    expect(source).toContain("document.documentElement.setAttribute('data-layout', isMobile ? 'mobile' : 'desktop');");
   });
 });