fix: update state

pedronauck · Oct 6, 2018 · 862dfcc · 862dfcc
1 parent d7a6e9e
commit 862dfcc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/reworm.tsx b/src/reworm.tsx
@@ -60,8 +60,9 @@ export class Provider extends Component {
   }
 
   private handleUpdate = (id: string, next: any) => {
-    const nextState = typeof next === 'function' ? next(this.state) : next
-    this.setState({ [id]: nextState })
+    this.setState((prevState: any) => ({
+      [id]: typeof next === 'function' ? next(prevState[id]) : next,
+    }))
   }
 }