diff --git a/tests/db/conftest.py b/tests/db/conftest.py
index b55e1120..860b489b 100644
--- a/tests/db/conftest.py
+++ b/tests/db/conftest.py
@@ -10,7 +10,7 @@ import pytest
 import rethinkdb as r
 
 from bigchaindb import Bigchain
-from bigchaindb.backend import connect, init_database
+from bigchaindb.backend import connect, schema
 from bigchaindb.common import crypto
 from bigchaindb.common.exceptions import DatabaseAlreadyExists
 
@@ -34,7 +34,7 @@ def setup_database(request, node_config):
         conn.run(r.db_drop(db_name))
 
     try:
-        init_database()
+        schema.init_database()
     except DatabaseAlreadyExists:
         print('Database already exists.')
 
diff --git a/tests/db/rethinkdb/test_schema.py b/tests/db/rethinkdb/test_schema.py
index d4a24480..9c219ac9 100644
--- a/tests/db/rethinkdb/test_schema.py
+++ b/tests/db/rethinkdb/test_schema.py
@@ -4,7 +4,7 @@ import rethinkdb as r
 import bigchaindb
 from bigchaindb import backend
 from bigchaindb.backend.rethinkdb import schema
-from .conftest import setup_database as _setup_database
+from ..conftest import setup_database as _setup_database
 
 
 # Since we are testing database initialization and database drop,