diff --git a/test/test.js b/test/test.js
index 329cbb70..6ff1baa2 100644
--- a/test/test.js
+++ b/test/test.js
@@ -1,10 +1,12 @@
 'use strict';
 
 const fs = require('fs');
-const APNS = require('../lib/apns');
+const should = require('should');
+
+// Package
+const APNS = require('../');
 const errors = APNS.errors;
 const HTTP2Client = require('../lib/http2-client');
-const should = require('should');
 
 describe('http2', () => {