diff --git a/ui-tests/tests/replacePerItem.spec.ts b/ui-tests/tests/replacePerItem.spec.ts
index 69a3c5c..ccf0bf1 100644
--- a/ui-tests/tests/replacePerItem.spec.ts
+++ b/ui-tests/tests/replacePerItem.spec.ts
@@ -92,7 +92,7 @@ test('should replace results for a particular file only', async ({ page }) => {
   );
 });
 
-test.skip('should undo replace results for a particular file only', async ({
+test('should undo replace results for a particular file only', async ({
   page
 }) => {
   await page.getByRole('tab', { name: 'Search and Replace' }).click();
@@ -228,7 +228,7 @@ test('should replace results for a particular match only', async ({ page }) => {
   );
 });
 
-test.skip('should undo replace results for a particular match only', async ({
+test('should undo replace results for a particular match only', async ({
   page
 }) => {
   await page.getByRole('tab', { name: 'Search and Replace' }).click();
diff --git a/ui-tests/tests/search.spec.ts b/ui-tests/tests/search.spec.ts
index 65f07b5..d23cce3 100644
--- a/ui-tests/tests/search.spec.ts
+++ b/ui-tests/tests/search.spec.ts
@@ -50,7 +50,7 @@ test('should get 5 matches', async ({ page, tmpPath }) => {
   );
 
   // Check the match is selected in the editor
-  await expect(page.locator('span.CodeMirror-selectedtext')).toHaveText(
+  await expect(page.locator('::selection')).toHaveText(
     'strange'
   );
 });