diff --git a/homeassistant/components/rainmachine/binary_sensor.py b/homeassistant/components/rainmachine/binary_sensor.py
index 78f3863dd16..bfd36ecf550 100644
--- a/homeassistant/components/rainmachine/binary_sensor.py
+++ b/homeassistant/components/rainmachine/binary_sensor.py
@@ -115,23 +115,26 @@ async def async_setup_entry(
         if api_category == DATA_PROVISION_SETTINGS:
             return partial(
                 ProvisionSettingsBinarySensor,
+                entry,
                 coordinators[DATA_PROVISION_SETTINGS],
             )
 
         if api_category == DATA_RESTRICTIONS_CURRENT:
             return partial(
                 CurrentRestrictionsBinarySensor,
+                entry,
                 coordinators[DATA_RESTRICTIONS_CURRENT],
             )
 
         return partial(
             UniversalRestrictionsBinarySensor,
+            entry,
             coordinators[DATA_RESTRICTIONS_UNIVERSAL],
         )
 
     async_add_entities(
         [
-            async_get_sensor(description.api_category)(entry, controller, description)
+            async_get_sensor(description.api_category)(controller, description)
             for description in BINARY_SENSOR_DESCRIPTIONS
         ]
     )
diff --git a/homeassistant/components/rainmachine/sensor.py b/homeassistant/components/rainmachine/sensor.py
index 4eec124f936..beb893ddb74 100644
--- a/homeassistant/components/rainmachine/sensor.py
+++ b/homeassistant/components/rainmachine/sensor.py
@@ -104,17 +104,19 @@ async def async_setup_entry(
         if api_category == DATA_PROVISION_SETTINGS:
             return partial(
                 ProvisionSettingsSensor,
+                entry,
                 coordinators[DATA_PROVISION_SETTINGS],
             )
 
         return partial(
             UniversalRestrictionsSensor,
+            entry,
             coordinators[DATA_RESTRICTIONS_UNIVERSAL],
         )
 
     async_add_entities(
         [
-            async_get_sensor(description.api_category)(entry, controller, description)
+            async_get_sensor(description.api_category)(controller, description)
             for description in SENSOR_DESCRIPTIONS
         ]
     )