From f3b3005ef4eb1c47c3d483fa38ef6f3bbc60f460 Mon Sep 17 00:00:00 2001
From: chnliquan <chnliquan@163.com>
Date: Wed, 14 Apr 2021 20:40:15 +0800
Subject: [PATCH] fix: immer return value

---
 docs/demos/counter/hooks/model.ts | 1 -
 src/core/Model.tsx                | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/docs/demos/counter/hooks/model.ts b/docs/demos/counter/hooks/model.ts
index 28ea015..b36f7b8 100644
--- a/docs/demos/counter/hooks/model.ts
+++ b/docs/demos/counter/hooks/model.ts
@@ -26,4 +26,3 @@ export const counter = createModel<RootModel, 'counter'>()({
     },
   }),
 })
-
diff --git a/src/core/Model.tsx b/src/core/Model.tsx
index 9bff22f..79ae568 100644
--- a/src/core/Model.tsx
+++ b/src/core/Model.tsx
@@ -145,7 +145,7 @@ export class Model<C extends ModelConfig> {
 
     if (typeof state === 'object') {
       newState = produce(state, draft => {
-        reducer(draft, ...payload)
+        return reducer(draft, ...payload)
       })
     } else {
       newState = reducer(state, ...payload)