diff --git a/infrahub/generators/datacenter_query.gql b/infrahub/generators/datacenter_query.gql
index d8f630e..634d20d 100644
--- a/infrahub/generators/datacenter_query.gql
+++ b/infrahub/generators/datacenter_query.gql
@@ -1,7 +1,7 @@
 # GraphQL query to fetch InfraDatacenter with all required attributes
 # This query retrieves all necessary data for the datacenter generator
 
-query DatacenterQuery($datacenter_id: String!) {
+query DatacenterQuery($datacenter_id: ID!) {
   InfraDatacenter(ids: [$datacenter_id]) {
     edges {
       node {
@@ -52,4 +52,4 @@ query DatacenterQuery($datacenter_id: String!) {
       }
     }
   }
-}
+}
\ No newline at end of file