From d1a7eb88bd5e1e9a5ad37c40005b6eba0c233674 Mon Sep 17 00:00:00 2001
From: Sam Rose <hello@samwho.dev>
Date: Tue, 28 Jan 2025 12:01:58 +0000
Subject: [PATCH] Fix automation test modal row select losing track of the
 selected row.

---
 .../automation/SetupPanel/AutomationBlockSetup.svelte      | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/builder/src/components/automation/SetupPanel/AutomationBlockSetup.svelte b/packages/builder/src/components/automation/SetupPanel/AutomationBlockSetup.svelte
index a6ba1b2e44..14b3ae082b 100644
--- a/packages/builder/src/components/automation/SetupPanel/AutomationBlockSetup.svelte
+++ b/packages/builder/src/components/automation/SetupPanel/AutomationBlockSetup.svelte
@@ -424,7 +424,7 @@
                 autocomplete: true,
                 filter: false,
                 getOptionLabel: row => row?.[primaryDisplay] || "",
-                compare: (a, b) => a?._id === b?._id,
+                compare: (a, b) => a?.[primaryDisplay] === b?.[primaryDisplay],
                 onChange: e => {
                   if (isTestModal) {
                     onChange({
@@ -651,7 +651,10 @@
           ...request,
         }
 
-        if (newTestData?.row?._id == null) {
+        if (
+          newTestData?.row == null ||
+          Object.keys(newTestData?.row).length === 0
+        ) {
           selectedRow = null
         } else {
           selectedRow = newTestData.row