diff --git a/packages/lit-dev-tests/src/playwright/util.ts b/packages/lit-dev-tests/src/playwright/util.ts
index 3dfd63de5..37a9ca973 100644
--- a/packages/lit-dev-tests/src/playwright/util.ts
+++ b/packages/lit-dev-tests/src/playwright/util.ts
@@ -13,7 +13,7 @@ export const preventGDPRBanner = async (page: Page) => {
 };
 
 export const waitForTheme = async (page: Page, dark: boolean) => {
-  await page.waitForSelector('body.auto');
+  await page.waitForSelector('body.auto,body.light,body.dark');
 
   if (dark) {
     await page.locator('body').evaluate((body) => {