diff --git a/tests/dbus_middleware1/bus_provider.py b/tests/dbus_middleware1/bus_provider.py
index 0f753d8..cc69660 100644
--- a/tests/dbus_middleware1/bus_provider.py
+++ b/tests/dbus_middleware1/bus_provider.py
@@ -3,6 +3,8 @@
 # SPDX-License-Identifier: GPL-2.0-or-later
 from time import sleep
 
+from gi.repository.Gio import dbus_address_get_for_bus_sync, BusType
+
 from tests.dbus_middleware1.fake_devices import PiControlDeviceMockup
 
 
@@ -12,12 +14,13 @@ class TestBusProvider(PiControlDeviceMockup):
         super().setUp()
 
         # Do not import things on top of the module. Some classes or functions need to be mocked up first.
-        from revpi_middleware.dbus_middleware1 import BusProvider
+        from revpi_middleware.dbus_middleware1 import BusProviderMiddleware1
 
         # Prepare the bus provider and start it
-        self.bp = BusProvider(
-            self.picontrol.name,
-            use_system_bus=False,
+        bus_address = dbus_address_get_for_bus_sync(BusType.SESSION)
+        self.bp = BusProviderMiddleware1(
+            bus_address,
+            picontrol_device=self.picontrol.name,
         )
         self.bp.start()