diff --git a/src/notifications/notification.entity.ts b/src/notifications/notification.entity.ts
index 3df76ac1847e6efa32b346142820628774f2b230..f9d5dca9290c30da406a63adf56ee06fe716a7b5 100644
--- a/src/notifications/notification.entity.ts
+++ b/src/notifications/notification.entity.ts
@@ -12,11 +12,12 @@ import { GameEntity } from '../game/game.entity';
 @Entity('Notifications')
 export class NotificationEntity {
   @PrimaryGeneratedColumn('uuid') id: string;
+  @Column('text') type: string;
   @Column({ type: 'text' }) message: string;
   @CreateDateColumn() issued: Date;
 
   @ManyToOne(type => GameEntity, game => game.id, {
     onDelete: 'CASCADE',
   })
-  game: GameEntity;
+  game: string;
 }