diff --git a/src/controllers/pharmacy/unit.ts b/src/controllers/pharmacy/unit.ts
index e48b2cd..8b94b3d 100644
--- a/src/controllers/pharmacy/unit.ts
+++ b/src/controllers/pharmacy/unit.ts
@@ -179,6 +179,8 @@ export class UnitController {
 
             const data = new Unit()
             data.unit_name = param.unit_name
+            data.small_unit = param.small_unit
+            data.large_unit = param.large_unit
             data.created_by = req.auth.data.name
 
             await OrmHelper.DB.manager.save(data);
@@ -233,6 +235,8 @@ export class UnitController {
 
             if (data != null) {
                 data.unit_name = param.unit_name
+                data.small_unit = param.small_unit
+                data.large_unit = param.large_unit
                 data.updated_by = req.auth.data.name
 
                 await repo.save(data);
diff --git a/src/swagger/builder.js b/src/swagger/builder.js
index de99f7d..cd0888a 100644
--- a/src/swagger/builder.js
+++ b/src/swagger/builder.js
@@ -222,6 +222,8 @@ const doc = {
             },
             unit: {
                 $unit_name: 'Unit name',
+                $small_unit: 'Small unit name',
+                $large_unit: 'Large unit name'
             },
             item_category: {
                 $category_name: 'ItemCategory name',