diff --git a/tests/components/control4/test_config_flow.py b/tests/components/control4/test_config_flow.py
index d1faf2da6c6..9a1b392f61c 100644
--- a/tests/components/control4/test_config_flow.py
+++ b/tests/components/control4/test_config_flow.py
@@ -20,25 +20,23 @@ from homeassistant.data_entry_flow import FlowResultType
 from tests.common import MockConfigEntry
 
 
-def _get_mock_c4_account(
-    getAccountControllers={
+def _get_mock_c4_account():
+    c4_account_mock = AsyncMock(C4Account)
+
+    c4_account_mock.getAccountControllers.return_value = {
         "controllerCommonName": "control4_model_00AA00AA00AA",
         "href": "https://apis.control4.com/account/v3/rest/accounts/000000",
         "name": "Name",
-    },
-    getDirectorBearerToken={"token": "token"},
-):
-    c4_account_mock = AsyncMock(C4Account)
+    }
 
-    c4_account_mock.getAccountControllers.return_value = getAccountControllers
-    c4_account_mock.getDirectorBearerToken.return_value = getDirectorBearerToken
+    c4_account_mock.getDirectorBearerToken.return_value = {"token": "token"}
 
     return c4_account_mock
 
 
-def _get_mock_c4_director(getAllItemInfo={}):
+def _get_mock_c4_director():
     c4_director_mock = AsyncMock(C4Director)
-    c4_director_mock.getAllItemInfo.return_value = getAllItemInfo
+    c4_director_mock.getAllItemInfo.return_value = {}
 
     return c4_director_mock