diff --git a/hassio/dbus/utils.py b/hassio/dbus/utils.py index 9553927e7..b8656a235 100644 --- a/hassio/dbus/utils.py +++ b/hassio/dbus/utils.py @@ -5,10 +5,10 @@ from ..exceptions import DBusNotConnectedError def dbus_connected(method): """Wrapper for check if dbus is connected.""" - def wrap_dbus(self, *args, **kwargs): + def wrap_dbus(api, *args, **kwargs): """Check if dbus is connected before call a method.""" - if self.dbus is None: - raise DBusNotConnectedError(f"{self!s} not connected to dbus!") - return self.method(*args, **kwargs) + if api.dbus is None: + raise DBusNotConnectedError(f"{api!s} not connected to dbus!") + return method(api, *args, **kwargs) return wrap_dbus