diff --git a/api/v1/openapi.yaml b/api/v1/openapi.yaml
index 2408213..f9d442f 100644
--- a/api/v1/openapi.yaml
+++ b/api/v1/openapi.yaml
@@ -121,6 +121,9 @@ components:
       schema:
         type: string
         format: uuid
+        x-go-type: training.TrainingID
+        x-go-type-import:
+          path: gitlab.mareshq.com/hq/yggdrasil/pkg/training
       description: Training ID
 
   schemas:
@@ -164,6 +167,9 @@ components:
             id:
               type: string
               format: uuid
+              x-go-type: training.TrainingID
+              x-go-type-import:
+                path: gitlab.mareshq.com/hq/yggdrasil/pkg/training
         - required:
             - id