diff --git a/api/index.js b/api/index.js
index 79beace..ee798f7 100644
--- a/api/index.js
+++ b/api/index.js
@@ -36,10 +36,12 @@ fi`;
 	}, usage);
 };
 
-module.exports = reverseShell;
-
-module.exports.handler = (request, response) => {
+const handler = (request, response) => {
 	const { address } = request.query;
 
 	response.send(reverseShell(address));
 };
+
+module.exports = handler;
+
+module.exports.reverseShell = reverseShell;
diff --git a/test/test.js b/test/test.js
index 1b7ee6e..f6e2f65 100644
--- a/test/test.js
+++ b/test/test.js
@@ -1,5 +1,5 @@
 import test from 'ava';
-import reverseShell from 'this';
+import { reverseShell } from 'this';
 
 test('reverseShell is a function', t => {
 	t.is(typeof reverseShell, 'function');