diff --git a/sdbus_async/systemd/objects.py b/sdbus_async/systemd/objects.py index d2b04c7..08ca01e 100644 --- a/sdbus_async/systemd/objects.py +++ b/sdbus_async/systemd/objects.py @@ -22,7 +22,11 @@ def __init__(self, bus: Optional[SdBus] = None) -> None: async def get_unit_obj(self, name: str) -> SystemdUnit: object_path = await self.get_unit(name) - return SystemdUnit(object_path) + return SystemdUnit(object_path, self._attached_bus) + + async def load_unit_obj(self, name: str) -> SystemdUnit: + object_path = await self.load_unit(name) + return SystemdUnit(object_path, self._attached_bus) class SystemdUnit(SystemdUnitInterfaceAsync): diff --git a/sdbus_block/systemd/objects.py b/sdbus_block/systemd/objects.py index 6f5f3f2..b49b546 100644 --- a/sdbus_block/systemd/objects.py +++ b/sdbus_block/systemd/objects.py @@ -23,7 +23,11 @@ def __init__(self, bus: Optional[SdBus] = None) -> None: def get_unit_obj(self, name: str) -> SystemdUnit: object_path = self.get_unit(name) - return SystemdUnit(object_path) + return SystemdUnit(object_path, self._attached_bus) + + def load_unit_obj(self, name: str) -> SystemdUnit: + object_path = self.load_unit(name) + return SystemdUnit(object_path, self._attached_bus) class SystemdUnit(SystemdUnitInterface):