diff --git a/apps/edr-freight-api/src/config/database.config.ts b/apps/edr-freight-api/src/config/database.config.ts
index bbac17e7a..a511c1654 100644
--- a/apps/edr-freight-api/src/config/database.config.ts
+++ b/apps/edr-freight-api/src/config/database.config.ts
@@ -17,7 +17,7 @@ import {
   PositionType,
   Position,
   Project,
-  UnitConfiguration,
+  UnitSetting,
   GlobalUnitConfiguration,
   Unit,
   EmployeeSignature,
@@ -64,7 +64,7 @@ const iamEntities = [
   PositionType,
   Position,
   Project,
-  UnitConfiguration,
+  UnitSetting,
   GlobalUnitConfiguration,
   Unit,
   EmployeeSignature,