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);
       });
     });