handle async funcs in control flow methods

1 files changed, 2 insertions, 1 deletions

diff --git a/lib/auto.js b/lib/auto.js
index fe98e06..4d7bf32 100644
--- a/lib/auto.js
+++ b/lib/auto.js
@@ -8,6 +8,7 @@ import rest from './internal/rest';
 
 import once from './internal/once';
 import onlyOnce from './internal/onlyOnce';
+import wrapAsync from './internal/wrapAsync';
 
 /**
  * Determines the best order for running the functions in `tasks`, based on
@@ -213,7 +214,7 @@ export default function (tasks, concurrency, callback) {
         }));
 
         runningTasks++;
-        var taskFn = task[task.length - 1];
+        var taskFn = wrapAsync(task[task.length - 1]);
         if (task.length > 1) {
             taskFn(results, taskCallback);
         } else {