diff --git a/test/mock4js-core-test.js b/test/mock4js-core-test.js
index 4323630..2d65109 100644
--- a/test/mock4js-core-test.js
+++ b/test/mock4js-core-test.js
@@ -1,6 +1,6 @@
-var Mock = require('..').Mock;
-var Mock4JS = require('..').Mock4JS;
-var Mock4JSException = require('..').Mock4JSException;
+var Mock4JS = require('..');
+var Mock = require('../src/mock4js').Mock;
+var Mock4JSException = require('../src/mock4js').Mock4JSException;
 var expect = require('expectations');
 
 describe('Mock4JS core', function() {
diff --git a/test/mock4js-messages.js b/test/mock4js-messages.js
index 7b284fa..8d19068 100644
--- a/test/mock4js-messages.js
+++ b/test/mock4js-messages.js
@@ -1,6 +1,4 @@
-var Mock = require('..').Mock;
-var Mock4JS = require('..').Mock4JS;
-var Mock4JSException = require('..').Mock4JSException;
+var Mock4JS = require('..');
 var expect = require('expectations');
 
 describe('Mock4JS messages', function() {