From 2edad82078335760261b6d8bcc4e43e2b0c08ade Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Sun, 17 Oct 2021 05:44:48 -1000
Subject: [PATCH] Fix order of arguments in rainmachine sensors (#57895)

---
 homeassistant/components/rainmachine/binary_sensor.py | 5 ++++-
 homeassistant/components/rainmachine/sensor.py        | 4 +++-
 2 files changed, 7 insertions(+), 2 deletions(-)

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
         ]
     )