From f2951fb51fa338da7a651cf414a074f2acf8b4cb Mon Sep 17 00:00:00 2001
From: Dan <dan.abramov@gmail.com>
Date: Sun, 7 Apr 2019 17:40:53 +0100
Subject: [PATCH] Preserve selection on exiting owner mode

---
 src/devtools/views/Components/TreeContext.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/devtools/views/Components/TreeContext.js b/src/devtools/views/Components/TreeContext.js
index 940821c1a1..b83078f653 100644
--- a/src/devtools/views/Components/TreeContext.js
+++ b/src/devtools/views/Components/TreeContext.js
@@ -383,7 +383,10 @@ function reduceOwnersState(store: Store, state: State, action: Action): State {
     case 'RESET_OWNER_STACK':
       ownerStack = [];
       ownerStackIndex = null;
-      selectedElementIndex = null;
+      selectedElementIndex =
+        selectedElementID !== null
+          ? store.getIndexOfElementID(selectedElementID)
+          : null;
       _ownerFlatTree = null;
       break;
     case 'SELECT_ELEMENT_AT_INDEX':