2 files changed, 1 insertions, 4 deletions
diff --git a/tests/mocks/database/db/driver.php b/tests/mocks/database/db/driver.php
index 2cf54b97b..86f7efd54 100644
--- a/tests/mocks/database/db/driver.php
+++ b/tests/mocks/database/db/driver.php
@@ -37,4 +37,4 @@ class Mock_Database_DB_Driver extends CI_DB_driver {
 
 }
 
-class CI_DB extends Mock_Database_DB_QueryBuilder {}
-\ No newline at end of file
+class CI_DB extends CI_DB_query_builder {}
+\ No newline at end of file
diff --git a/tests/mocks/database/db/querybuilder.php b/tests/mocks/database/db/querybuilder.php
deleted file mode 100644
index 3f2252622..000000000
--- a/tests/mocks/database/db/querybuilder.php
+++ /dev/null
@@ -1,3 +0,0 @@
-<?php
-
-class Mock_Database_DB_QueryBuilder extends CI_DB_query_builder {}
-\ No newline at end of file