diff --git a/tests/components/mochad/test_switch.py b/tests/components/mochad/test_switch.py
index 9fea3b5c14c..d7875246fac 100644
--- a/tests/components/mochad/test_switch.py
+++ b/tests/components/mochad/test_switch.py
@@ -9,6 +9,8 @@ from homeassistant.components.mochad import switch as mochad
 from homeassistant.core import HomeAssistant
 from homeassistant.setup import async_setup_component
 
+from tests.common import MockEntityPlatform
+
 
 @pytest.fixture(autouse=True)
 def pymochad_mock():
@@ -25,7 +27,9 @@ def switch_mock(hass: HomeAssistant) -> mochad.MochadSwitch:
     """Mock switch."""
     controller_mock = mock.MagicMock()
     dev_dict = {"address": "a1", "name": "fake_switch"}
-    return mochad.MochadSwitch(hass, controller_mock, dev_dict)
+    entity = mochad.MochadSwitch(hass, controller_mock, dev_dict)
+    entity.platform = MockEntityPlatform(hass)
+    return entity
 
 
 async def test_setup_adds_proper_devices(hass: HomeAssistant) -> None: