diff --git a/index.js b/index.js
index 956d31b9..c849218c 100644
--- a/index.js
+++ b/index.js
@@ -1,9 +1,9 @@
 // For development/testing purposes
-exports.handler = function( event, context ) {
-  console.log( "Running index.handler" );
-  console.log( "==================================");
-  console.log( "event", event );
-  console.log( "==================================");
-  console.log( "Stopping index.handler" );
-  context.done( );
+exports.handler = function(event, context, callback) {
+  console.log('Running index.handler');
+  console.log('==================================');
+  console.log('event', event);
+  console.log('==================================');
+  console.log('Stopping index.handler');
+  callback(null);
 };