new build

jurgob · Jun 14, 2016 · 7188c73 · 7188c73
1 parent dda8cec
commit 7188c73
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -73,11 +73,18 @@ var set = exports.set = function set(target) {
 };
 var update = exports.update = function update(target, updateCB) {
   return function (state, action) {
-    return updateIn(state, target, function (oldValue) {
+    var updateFinalCb = void 0;
+
+    if (typeof updateCB === 'function') updateFinalCb = function updateFinalCb(oldValue) {
       return updateCB(oldValue, action.payload);
-    });
+    };else updateFinalCb = function updateFinalCb() {
+      return updateCB;
+    };
+
+    return updateIn(state, target, updateFinalCb);
   };
 };
+
 var deepmerge = exports.deepmerge = function deepmerge(target) {
   return function (state, action) {
     return updateIn(state, target, function (obj) {