diff --git a/src/e2e.spec.js b/src/e2e.spec.js
index 6ac390f..cc6557c 100644
--- a/src/e2e.spec.js
+++ b/src/e2e.spec.js
@@ -1,8 +1,7 @@
 import {expect} from 'chai';
-import {graphql} from 'graphql';
 import mongoose from 'mongoose';
 
-import {getSchema} from './schema';
+import {getSchema, graphql} from './schema';
 import User from '../fixture/user';
 
 describe('e2e', () => {
diff --git a/src/schema.spec.js b/src/schema.spec.js
index 252532e..8a43c9e 100644
--- a/src/schema.spec.js
+++ b/src/schema.spec.js
@@ -1,9 +1,8 @@
 import {expect} from 'chai';
-import {graphql} from 'graphql';
 import mongoose from 'mongoose';
 import ObjectID from 'bson-objectid';
 
-import {getSchema} from './schema';
+import {getSchema, graphql} from './schema';
 import User from '../fixture/user';
 
 describe('schema', () => {