diff --git a/src/store/useTabsStore.ts b/src/store/useTabsStore.ts
index 9ef48f8..b40fa7a 100644
--- a/src/store/useTabsStore.ts
+++ b/src/store/useTabsStore.ts
@@ -22,15 +22,17 @@ export const useTabsStore = create<TabsState>()(
     }),
     {
       name: "tabs-active",
-      onRehydrateStorage:
-        () =>
-          (state, error) => {
-            if (error) {
-              console.error("hydrate error", error);
-            } else if (state) {
-              state.setHydrated(true);
-            }
-          },
+      partialize: (state) => ({
+        problemTab: state.problemTab,
+        workspaceTab: state.workspaceTab,
+      }),
+      onRehydrateStorage: () => (state, error) => {
+        if (error) {
+          console.error("hydrate error", error);
+        } else if (state) {
+          state.setHydrated(true);
+        }
+      },
     }
   )
 );