diff --git a/test/relations.test.js b/test/relations.test.js
index a7954846..b99f31ba 100644
--- a/test/relations.test.js
+++ b/test/relations.test.js
@@ -1,7 +1,7 @@
 // This test written in mocha+should.js
 var should = require('./init.js');
 var jdb = require('../');
-var Schema = jdb.Schema;
+var DataSource = jdb.DataSource;
 
 var db, tmp, Book, Chapter, Author, Reader;
 var Category, Job;
@@ -10,7 +10,7 @@ var Person, Address;
 var Link;
 
 var getTransientDataSource = function(settings) {
-    return new Schema('transient', settings, db.modelBuilder);
+    return new DataSource('transient', settings, db.modelBuilder);
 };
 
 describe('relations', function () {
diff --git a/test/transient.test.js b/test/transient.test.js
index d6ad1d3e..e3a03f96 100644
--- a/test/transient.test.js
+++ b/test/transient.test.js
@@ -1,5 +1,4 @@
 var jdb = require('../');
-var Schema = jdb.Schema;
 var DataSource = jdb.DataSource;
 var assert = require('assert');
 var async = require('async');
@@ -8,7 +7,7 @@ var should = require('./init.js');
 var db, TransientModel, Person, Widget, Item;
 
 var getTransientDataSource = function(settings) {
-    return new Schema('transient', settings);
+    return new DataSource('transient', settings);
 };
 
 describe('Transient connector', function () {