diff --git a/src/pages/client/pages/ClientInstancePage/modals/DeleteInstanceModal/DeleteInstanceModal.tsx b/src/pages/client/pages/ClientInstancePage/modals/DeleteInstanceModal/DeleteInstanceModal.tsx
index f743919c..fb532e3b 100644
--- a/src/pages/client/pages/ClientInstancePage/modals/DeleteInstanceModal/DeleteInstanceModal.tsx
+++ b/src/pages/client/pages/ClientInstancePage/modals/DeleteInstanceModal/DeleteInstanceModal.tsx
@@ -49,7 +49,6 @@ export const DeleteInstanceModal = () => {
           refetchType: 'active',
         });
       });
-      setClientStore({ selectedInstance: undefined, selectedLocation: undefined });
       close();
       closeUpdate();
     },
@@ -83,6 +82,7 @@ export const DeleteInstanceModal = () => {
       cancelText={LL.common.controls.cancel()}
       onSubmit={() => {
         if (instance) {
+          setClientStore({ selectedInstance: undefined, selectedLocation: undefined });
           mutate(instance.id);
         }
       }}
diff --git a/src/pages/client/pages/ClientInstancePage/modals/UpdateInstanceModal/components/UpdateInstanceModalForm.tsx b/src/pages/client/pages/ClientInstancePage/modals/UpdateInstanceModal/components/UpdateInstanceModalForm.tsx
index 1b684680..cae2a283 100644
--- a/src/pages/client/pages/ClientInstancePage/modals/UpdateInstanceModal/components/UpdateInstanceModalForm.tsx
+++ b/src/pages/client/pages/ClientInstancePage/modals/UpdateInstanceModal/components/UpdateInstanceModalForm.tsx
@@ -4,7 +4,6 @@ import { fetch } from '@tauri-apps/plugin-http';
 import { useMemo } from 'react';
 import { type SubmitHandler, useForm } from 'react-hook-form';
 import { z } from 'zod';
-import { shallow } from 'zustand/shallow';
 import { useI18nContext } from '../../../../../../../i18n/i18n-react';
 import { FormInput } from '../../../../../../../shared/defguard-ui/components/Form/FormInput/FormInput';
 import { Button } from '../../../../../../../shared/defguard-ui/components/Layout/Button/Button';
@@ -40,7 +39,6 @@ export const UpdateInstanceModalForm = () => {
   const closeModal = useUpdateInstanceModal((state) => state.close);
   const toaster = useToaster();
   const queryClient = useQueryClient();
-  const setClientState = useClientStore((s) => s.setState, shallow);
   const platformInfo = useClientStore((state) => state.platformInfo);
 
   const defaultValues = useMemo(
@@ -226,10 +224,6 @@ export const UpdateInstanceModalForm = () => {
           text={localLL.controls.removeInstance()}
           onClick={() => {
             if (instance) {
-              setClientState({
-                selectedInstance: undefined,
-                selectedLocation: undefined,
-              });
               openDeleteInstance(instance);
             }
           }}