diff --git a/gso/workflows/device/create_device.py b/gso/workflows/device/create_device.py
index 08b204b565b2ddac96ef5050d45335f4d366e03d..681b52590d88d856b4d64373b53332c9191e43ed 100644
--- a/gso/workflows/device/create_device.py
+++ b/gso/workflows/device/create_device.py
@@ -30,7 +30,7 @@ from gso.services.provisioning_proxy import (
 )
 
 
-def site_selector() -> list:
+def site_selector() -> Choice:
     site_subscriptions = {}
     for site_id, site_description in (
         SubscriptionTable.query.join(ProductTable)
@@ -131,7 +131,7 @@ def initialize_subscription(
     subscription.device.device_ts_port = str(ts_port)
     subscription.device.device_vendor = device_vendor
     subscription.device.device_site = Site.from_subscription(
-        device_site[0]
+        device_site
     ).site
     fqdn = (
         f"{hostname}.{subscription.device.device_site.site_name.lower()}."