diff --git a/playwriter/src/snapshot-tools.test.ts b/playwriter/src/snapshot-tools.test.ts
index 074b839..0ca38a5 100644
--- a/playwriter/src/snapshot-tools.test.ts
+++ b/playwriter/src/snapshot-tools.test.ts
@@ -633,7 +633,7 @@ describe('Snapshot & Screenshot Tests', () => {
         console.log('Button selector:', btnSelector)
 
         // Verify the selector works
-        const btnViaSelector = cdpPage!.locator('[data-testid="submit-btn"]')
+        const btnViaSelector = cdpPage!.locator(btnSelector!)
         const btnTextViaRef = await btnViaSelector.textContent()
         console.log('Button text via selector:', btnTextViaRef)
         expect(btnTextViaRef).toBe('Submit Form')