diff --git a/homeassistant/components/prosegur/config_flow.py b/homeassistant/components/prosegur/config_flow.py index ea975529b01..ac2b704b012 100644 --- a/homeassistant/components/prosegur/config_flow.py +++ b/homeassistant/components/prosegur/config_flow.py @@ -21,7 +21,9 @@ STEP_USER_DATA_SCHEMA = vol.Schema( { vol.Required(CONF_USERNAME): str, vol.Required(CONF_PASSWORD): str, - vol.Required(CONF_COUNTRY): vol.In(COUNTRY.keys()), + vol.Required(CONF_COUNTRY): selector.CountrySelector( + selector.CountrySelectorConfig(countries=list(COUNTRY)) + ), } )