diff --git a/tests/src/Functional/IpAddressBlockingTest.php b/tests/src/Functional/IpAddressBlockingTest.php
index 0e39c8c..d0c3334 100644
--- a/tests/src/Functional/IpAddressBlockingTest.php
+++ b/tests/src/Functional/IpAddressBlockingTest.php
@@ -119,7 +119,7 @@ class IpAddressBlockingTest extends BrowserTestBase {
     $banIp->banIp($ip);
     $banIp->banIp($ip);
     $banIp->banIp($ip);
-    $query = db_select('advban_ip', 'bip');
+    $query = \Drupal::database()->select('advban_ip', 'bip');
     $query->fields('bip', ['iid']);
     $query->condition('bip.ip', $ip);
     $ip_count = $query->execute()->fetchAll();
@@ -128,7 +128,7 @@ class IpAddressBlockingTest extends BrowserTestBase {
     $ip = '';
     $banIp->banIp($ip);
     $banIp->banIp($ip);
-    $query = db_select('advban_ip', 'bip');
+    $query = \Drupal::database()->select('advban_ip', 'bip');
     $query->fields('bip', ['iid']);
     $query->condition('bip.ip', $ip);
     $ip_count = $query->execute()->fetchAll();