diff --git a/tests/components/modbus/conftest.py b/tests/components/modbus/conftest.py index 7f67a0653fb..43221b219e5 100644 --- a/tests/components/modbus/conftest.py +++ b/tests/components/modbus/conftest.py @@ -97,12 +97,9 @@ async def base_test( mock_sync = mock.MagicMock() with mock.patch( - "homeassistant.components.modbus.modbus.ModbusTcpClient", return_value=mock_sync - ), mock.patch( - "homeassistant.components.modbus.modbus.ModbusSerialClient", + "homeassistant.components.modbus.modbus.ModbusTcpClient", + autospec=True, return_value=mock_sync, - ), mock.patch( - "homeassistant.components.modbus.modbus.ModbusUdpClient", return_value=mock_sync ): # Setup inputs for the sensor