From e33c44c6cc5ec66e62ea52ca655815beb4478560 Mon Sep 17 00:00:00 2001
From: Janne Savolainen <janne.savolainen@live.fi>
Date: Mon, 20 Jun 2022 10:21:28 +0300
Subject: [PATCH] Consolidate stuff under same MobX action

Co-authored-by: Mikko Aspiala <mikko.aspiala@gmail.com>

Signed-off-by: Janne Savolainen <janne.savolainen@live.fi>
---
 .../components/test-utils/get-application-builder.tsx         | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/renderer/components/test-utils/get-application-builder.tsx b/src/renderer/components/test-utils/get-application-builder.tsx
index 32a4bba41f..ae54d4f71c 100644
--- a/src/renderer/components/test-utils/get-application-builder.tsx
+++ b/src/renderer/components/test-utils/get-application-builder.tsx
@@ -543,9 +543,9 @@ const disableExtensionsFor = <T extends ObservableSet>(
     extensionInstances.forEach((instance) => {
       const extension = getExtension(instance);
 
-      extension.deregister();
-
       runInAction(() => {
+        extension.deregister();
+
         extensionState.delete(extension);
       });
     });