diff --git a/tests/components/derivative/test_init.py b/tests/components/derivative/test_init.py
index 32b763ee84d..0081ab97580 100644
--- a/tests/components/derivative/test_init.py
+++ b/tests/components/derivative/test_init.py
@@ -62,10 +62,12 @@ async def test_setup_and_remove_config_entry(
     assert entity_registry.async_get(derivative_entity_id) is None
 
 
-async def test_device_cleaning(hass: HomeAssistant) -> None:
+async def test_device_cleaning(
+    hass: HomeAssistant,
+    device_registry: dr.DeviceRegistry,
+    entity_registry: er.EntityRegistry,
+) -> None:
     """Test for source entity device for Derivative."""
-    device_registry = dr.async_get(hass)
-    entity_registry = er.async_get(hass)
 
     # Source entity device config entry
     source_config_entry = MockConfigEntry()