diff --git a/source/HomeAssistantMenuItemFactory.mc b/source/HomeAssistantMenuItemFactory.mc
index e5118f9..d18ba58 100644
--- a/source/HomeAssistantMenuItemFactory.mc
+++ b/source/HomeAssistantMenuItemFactory.mc
@@ -155,8 +155,8 @@ class HomeAssistantMenuItemFactory {
         label     as Lang.String or Lang.Symbol,
         entity_id as Lang.String?,
         template  as Lang.String?,
-        action   as Lang.String?,
-        data      as Lang.Dictionary?,
+        action    as Lang.String?,
+        picker    as Lang.Dictionary,
         options   as {
             :exit    as Lang.Boolean,
             :confirm as Lang.Boolean,