use initialParams to simplify callback extraction

1 files changed, 2 insertions, 4 deletions

diff --git a/lib/retryable.js b/lib/retryable.js
index 26fa609..f01d9b9 100644
--- a/lib/retryable.js
+++ b/lib/retryable.js
@@ -1,14 +1,12 @@
 import retry from './retry';
-import rest from 'lodash/rest';
+import initialParams from './internal/initialParams';
 
 export default function (opts, task) {
     if (!task) {
         task = opts;
         opts = null;
     }
-    return rest(function (args) {
-        var callback = args.pop();
-
+    return initialParams(function (args, callback) {
         function taskFn(cb) {
             task.apply(null, args.concat([cb]));
         }