diff --git a/tests/phpunit/OATHAuthModuleRegistryTest.php b/tests/phpunit/OATHAuthModuleRegistryTest.php
index 30aae400..b951bf48 100644
--- a/tests/phpunit/OATHAuthModuleRegistryTest.php
+++ b/tests/phpunit/OATHAuthModuleRegistryTest.php
@@ -30,7 +30,7 @@ class OATHAuthModuleRegistryTest extends MediaWikiIntegrationTestCase {
 	 * @covers \MediaWiki\Extension\OATHAuth\OATHAuthModuleRegistry::getModuleIds
 	 */
 	public function testGetModuleIds() {
-		$this->db->newInsertQueryBuilder()
+		$this->getDb()->newInsertQueryBuilder()
 			->insertInto( 'oathauth_types' )
 			->row( [ 'oat_name' => 'first' ] )
 			->caller( __METHOD__ )
diff --git a/tests/phpunit/integration/OATHUserRepositoryTest.php b/tests/phpunit/integration/OATHUserRepositoryTest.php
index 7f6dff0d..2ee02969 100644
--- a/tests/phpunit/integration/OATHUserRepositoryTest.php
+++ b/tests/phpunit/integration/OATHUserRepositoryTest.php
@@ -84,7 +84,7 @@ class OATHUserRepositoryTest extends MediaWikiIntegrationTestCase {
 		);
 
 		$this->assertNotEmpty(
-			$this->db->newSelectQueryBuilder()
+			$this->getDb()->newSelectQueryBuilder()
 				->select( '1' )
 				->from( 'oathauth_devices' )
 				->where( [ 'oad_user' => $oathUser->getCentralId() ] )