fix(useMap): remove now also uses prevMap state to not overwrite or l…

…ose state
streamich · Jun 27, 2019 · 0a59869 · 0a59869
1 parent 7ce8e4f
commit 0a59869
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/useMap.ts b/src/useMap.ts
@@ -21,8 +21,10 @@ const useMap = <T extends object = any>(initialMap: T = {} as T): [T, Actions<T>
         }));
       },
       remove: (key: keyof T) => {
-        const { [key]: omit, ...rest } = map;
-        set(rest as T);
+        set(prevMap => {
+          const { [key]: omit, ...rest } = prevMap;
+          return rest as T;
+        } );
       },
       reset: () => set(initialMap),
     },