diff --git a/backend/internal/data/ent/generate.go b/backend/internal/data/ent/generate.go
index eb03ded..7b8b727 100644
--- a/backend/internal/data/ent/generate.go
+++ b/backend/internal/data/ent/generate.go
@@ -1,3 +1,3 @@
 package ent
 
-//go:generate go run -mod=mod entgo.io/ent/cmd/ent generate --feature sql/versioned-migration ./schema
+//go:generate go run -mod=mod entgo.io/ent/cmd/ent generate --feature sql/versioned-migration ./schema --template=./schema/templates/has_id.tmpl
diff --git a/backend/internal/data/ent/schema/templates/has_id.tmpl b/backend/internal/data/ent/schema/templates/has_id.tmpl
index 42b0cd8..cc6e30a 100644
--- a/backend/internal/data/ent/schema/templates/has_id.tmpl
+++ b/backend/internal/data/ent/schema/templates/has_id.tmpl
@@ -9,8 +9,13 @@
 import 	"github.com/google/uuid"
 {{/* Loop over all nodes and implement the "HasID" interface */}}
 {{ range $n := $.Nodes }}
+    {{ if not $n.ID }}
+        {{/* If the node doesn't have an ID field, we skip it. */}}
+        {{ continue }}
+    {{ end }}
+    {{/* The "HasID" interface is implemented by the "ID" method. */}}
     {{ $receiver := $n.Receiver }}
-    func ({{ $receiver }} *{{ $n.Name }}) GetID() uuid.UUID {
+    func ({{ $receiver }} *{{ $n.Name }}) GetID() {{ $n.ID.Type }} {
         return {{ $receiver }}.ID
     }
 {{ end }}