diff --git a/tests/helpers/test_entity_component.py b/tests/helpers/test_entity_component.py index a95c4a6d0fd..1e12d7c3ea3 100644 --- a/tests/helpers/test_entity_component.py +++ b/tests/helpers/test_entity_component.py @@ -292,7 +292,7 @@ class TestHelpersEntityComponent(unittest.TestCase): assert platform2_setup.called @patch('homeassistant.helpers.entity_component.EntityComponent' - '._async_setup_platform') + '._async_setup_platform', return_value=mock_coro()()) @patch('homeassistant.bootstrap.async_setup_component', return_value=mock_coro(True)()) def test_setup_does_discovery(self, mock_setup_component, mock_setup):