diff --git a/libs/common/src/database/database.module.ts b/libs/common/src/database/database.module.ts
index ce51bae..bfda450 100644
--- a/libs/common/src/database/database.module.ts
+++ b/libs/common/src/database/database.module.ts
@@ -16,6 +16,8 @@ import { GroupDeviceEntity } from '../modules/group-device/entities';
 import { DeviceUserPermissionEntity } from '../modules/device-user-permission/entities';
 import { UserRoleEntity } from '../modules/user-role/entities';
 import { RoleTypeEntity } from '../modules/role-type/entities';
+import { UserNotificationEntity } from '../modules/user-notification/entities';
+import { DeviceNotificationEntity } from '../modules/device-notification/entities';
 
 @Module({
   imports: [
@@ -46,6 +48,8 @@ import { RoleTypeEntity } from '../modules/role-type/entities';
           DeviceUserPermissionEntity,
           UserRoleEntity,
           RoleTypeEntity,
+          UserNotificationEntity,
+          DeviceNotificationEntity,
         ],
         namingStrategy: new SnakeNamingStrategy(),
         synchronize: Boolean(JSON.parse(configService.get('DB_SYNC'))),