diff --git a/test/e2e/app-dir/not-found-default/index.test.ts b/test/e2e/app-dir/not-found-default/index.test.ts
index 01f5434d0e956..3ce29abfb4e97 100644
--- a/test/e2e/app-dir/not-found-default/index.test.ts
+++ b/test/e2e/app-dir/not-found-default/index.test.ts
@@ -35,7 +35,7 @@ createNextDescribe(
       const browser = await next.browser('/non-existent')
       await browser.waitForElementByCss('.next-error-h1')
       expect(await browser.elementByCss('.next-error-h1').text()).toBe('404')
-      expect(await browser.elementByCss('html').attr('class')).toBe(
+      expect(await browser.elementByCss('html').getAttribute('class')).toBe(
         'root-layout-html'
       )
     })