diff --git a/test/mocha-test-es6.mjs b/test/mocha-test-es6.mjs
index 54c4dca..c0bd217 100644
--- a/test/mocha-test-es6.mjs
+++ b/test/mocha-test-es6.mjs
@@ -1,5 +1,5 @@
-import { UAParser } from 'ua-parser-js';
-import { CPU, Device, Engine } from 'ua-parser-js/enums';
+import { UAParser } from '../src/main/ua-parser.mjs';
+import { CPU, Device, Engine } from '../src/enums/ua-parser-enums.mjs';
 import * as assert from 'assert';
 
 describe('Returns', () => {
diff --git a/test/mocha-test-extension.js b/test/mocha-test-extension.js
index fd29fd8..c338e3e 100644
--- a/test/mocha-test-extension.js
+++ b/test/mocha-test-extension.js
@@ -3,8 +3,8 @@ const assert = require('assert');
 const parseJS = require('@babel/parser').parse;
 const traverse = require('@babel/traverse').default;
 const safe = require('safe-regex');
-const UAParser = require('ua-parser-js');
-const { Bots, CLIs, Emails, Modules } = require('ua-parser-js/extensions');
+const UAParser = require('../src/main/ua-parser');
+const { Bots, CLIs, Emails, Modules } = require('../src/extensions/ua-parser-extensions');
 
 describe('Bots', () => {
     it('Can detect bots', () => {
diff --git a/test/mocha-test.js b/test/mocha-test.js
index 97c5867..6d6c937 100644
--- a/test/mocha-test.js
+++ b/test/mocha-test.js
@@ -4,7 +4,7 @@ var assert      = require('assert');
 var requirejs   = require('requirejs');
 var parseJS     = require('@babel/parser').parse;
 var traverse    = require('@babel/traverse').default;
-var UAParser    = require('ua-parser-js');
+var UAParser    = require('../src/main/ua-parser');
 var browsers    = require('./specs/browser-all.json');
 var cpus        = require('./specs/cpu-all.json');
 var devices     = require('./specs/device-all.json');