diff --git a/apex.js/apex.js-tests.ts b/apex.js/apex.js-tests.ts
index e42528d180..0595ef43ea 100644
--- a/apex.js/apex.js-tests.ts
+++ b/apex.js/apex.js-tests.ts
@@ -1,5 +1,8 @@
-import λ from "apex.js";
+import * as λ from "apex.js";
 
-const handler = λ(event => {
-    console.log("Invoked with event " + JSON.stringify(event));
+const handler = λ((event, context) => {
+    console.log("Event: " + JSON.stringify(event));
+    console.log("Context: " + JSON.stringify(context));
+
+    return {event, context};
 });
diff --git a/apex.js/index.d.ts b/apex.js/index.d.ts
index 54f70e1a7e..20178369dc 100644
--- a/apex.js/index.d.ts
+++ b/apex.js/index.d.ts
@@ -5,4 +5,6 @@
 
 /// <reference types="aws-lambda" />
 
-export default function λ(fn: (event: any, context: AWSLambda.Context) => any): (event: any, context: AWSLambda.Context, callback: AWSLambda.Callback) => void;
+declare function λ(fn: (event: any, context: AWSLambda.Context) => any): (event: any, context: AWSLambda.Context, callback: AWSLambda.Callback) => void;
+declare namespace λ {}
+export = λ;