diff --git a/homeassistant/components/nibe_heatpump/config_flow.py b/homeassistant/components/nibe_heatpump/config_flow.py
index f8e4974b79b..ba3325d9daf 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, default=9999): cv.port,
         vol.Required(CONF_REMOTE_READ_PORT, default=9999): cv.port,