diff --git a/src/index.js b/src/index.js index 2f3e8c4..877f646 100644 --- a/src/index.js +++ b/src/index.js @@ -65,10 +65,10 @@ function wrapHandler() { return iopipe(importedModule[handlerToWrap]); } -module.exports.handler = (event, context, callback) => { +module.exports.handler = function handler(...args) { if (!wrappedHandler) { wrappedHandler = wrapHandler(); } - return wrappedHandler(event, context, callback); + return wrappedHandler.apply(this, args); };