Fix for injection into final callback.

1 files changed, 17 insertions, 1 deletions

diff --git a/lib/autoInject.js b/lib/autoInject.js
index 9259487..245db8e 100644
--- a/lib/autoInject.js
+++ b/lib/autoInject.js
@@ -42,5 +42,21 @@ export default function autoInject(tasks, callback) {
         }
     });
 
-    auto(newTasks, callback);
+    auto(newTasks, function (err, results) {
+        var params;
+        if (isArray(callback)) {
+            params = clone(callback);
+            callback = params.pop();
+        } else {
+            params = parseParams(callback);
+            params.shift();
+        }
+
+        params = arrayMap(params, function (name) {
+            return results[name];
+        });
+
+        params.unshift(err);
+        callback.apply(null, params);
+    });
 }