diff --git a/tests/express.test.ts b/tests/express.test.ts
index 685e21a..a2c79fa 100644
--- a/tests/express.test.ts
+++ b/tests/express.test.ts
@@ -1,7 +1,7 @@
 import * as request from 'supertest-as-promised';
 import * as express from 'express';
 const redisMock = require('redis-mock');
-import { setLocalHeaders, setBasicSecurity, setBodyParser, setRateLimiter } from '../src/lib/express';
+import { setLocalHeaders, setBasicSecurity, setBodyParser, setRateLimiter } from '../src';
 
 describe('Express', () => {
   describe('#setLocalHeaders', () => {
diff --git a/tests/responder.test.ts b/tests/responder.test.ts
index 2c2c2e1..0225435 100644
--- a/tests/responder.test.ts
+++ b/tests/responder.test.ts
@@ -1,6 +1,6 @@
 import * as request from 'supertest-as-promised';
 import * as express from 'express';
-import * as responder from './../src/lib/responder';
+import * as responder from './../src';
 const app = express();
 
 describe('Responder', () => {
diff --git a/tests/server.test.ts b/tests/server.test.ts
index 56bb0b7..0028821 100644
--- a/tests/server.test.ts
+++ b/tests/server.test.ts
@@ -1,6 +1,6 @@
 import * as express from 'express';
 import * as request from 'supertest-as-promised';
-import { startServer } from '../src/lib/server';
+import { startServer } from '../src';
 
 // CONSTANTS
 const CONFIGURATION = {
diff --git a/tests/swagger.test.ts b/tests/swagger.test.ts
index 0ffa314..9d9ff5f 100644
--- a/tests/swagger.test.ts
+++ b/tests/swagger.test.ts
@@ -1,6 +1,6 @@
 import * as request from 'supertest-as-promised';
 import * as express from 'express';
-import { setSwagger } from './../src/lib/swagger';
+import { setSwagger } from './../src';
 const app = express();
 
 describe('Swagger', () => {