diff --git a/xbob/db/replay/test.py b/xbob/db/replay/test.py
index a3aebffdd99422ed214311da67d7b4e149642de5..a99664fcb549926ddd6baa4a86e935f15767b0e0 100644
--- a/xbob/db/replay/test.py
+++ b/xbob/db/replay/test.py
@@ -22,7 +22,7 @@
 
 import os, sys
 import unittest
-from . import db as replay_db
+from .query import Database
 
 class ReplayDatabaseTest(unittest.TestCase):
   """Performs various tests on the replay attack database."""
@@ -30,7 +30,7 @@ class ReplayDatabaseTest(unittest.TestCase):
   def test01_queryRealAccesses(self):
 
 
-    db = replay_db.Database()
+    db = Database()
     f = db.files(cls='real')
     self.assertEqual(len(set(f.values())), 200) #200 unique auth sessions
     for k,v in f.items():
@@ -53,7 +53,7 @@ class ReplayDatabaseTest(unittest.TestCase):
 
   def queryAttackType(self, protocol, N):
 
-    db = replay_db.Database()
+    db = Database()
     f = db.files(cls='attack', protocol=protocol)
 
     self.assertEqual(len(set(f.values())), N) 
@@ -99,7 +99,7 @@ class ReplayDatabaseTest(unittest.TestCase):
   
   def test08_queryEnrollments(self):
 
-    db = replay_db.Database()
+    db = Database()
     f = db.files(cls='enroll')
     self.assertEqual(len(set(f.values())), 100) #50 clients, 2 conditions
     for k,v in f.items():