Change Train DTO primary key type to a primitive one