diff --git a/src/faction/faction.dto.ts b/src/faction/faction.dto.ts
index 9064f56f285d3ab2aede368ad91c269ad8c2a334..fbc46d3267dc109010c4450a371ede7300c0d1d0 100644
--- a/src/faction/faction.dto.ts
+++ b/src/faction/faction.dto.ts
@@ -9,6 +9,7 @@ import {
   Max,
   IsOptional,
   IsHexColor,
+  IsIn,
 } from 'class-validator';
 
 import { GameEntity } from '../game/game.entity';
@@ -57,6 +58,8 @@ export class GameGroupDTO {
   @IsString()
   @Length(3, 31)
   name: string;
+  @IsIn(['infantry', 'recon', 'mechanized'])
+  class: string;
   @IsUUID('4')
   faction: FactionEntity;
 }