diff --git a/homeassistant/components/nibe_heatpump/config_flow.py b/homeassistant/components/nibe_heatpump/config_flow.py
index 28fafdb3a37..412b44d69a1 100644
--- a/homeassistant/components/nibe_heatpump/config_flow.py
+++ b/homeassistant/components/nibe_heatpump/config_flow.py
@@ -29,7 +29,7 @@ from .const import (
 
 STEP_USER_DATA_SCHEMA = vol.Schema(
     {
-        vol.Required(CONF_MODEL): vol.In([e.name for e in Model]),
+        vol.Required(CONF_MODEL): vol.In(list(Model.__members__)),
         vol.Required(CONF_IP_ADDRESS): str,
         vol.Required(CONF_LISTENING_PORT): cv.port,
         vol.Required(CONF_REMOTE_READ_PORT): cv.port,