diff --git a/internal/use-case/main-use-case/authentication/helper.go b/internal/use-case/main-use-case/authentication/helper.go
index c0a3bd0f..7af3afa5 100644
--- a/internal/use-case/main-use-case/authentication/helper.go
+++ b/internal/use-case/main-use-case/authentication/helper.go
@@ -226,6 +226,10 @@ func populateRoles(user *eu.User, input eu.LoginDto, atClaims jwt.MapClaims, out
 				outputData["doctor_code"] = doctor.Code
 
 				// specialist
+				if doctor.Unit_Code != nil {
+					atClaims["unit_code"] = doctor.Unit_Code
+					outputData["unit_code"] = doctor.Unit_Code
+				}
 				if doctor.Specialist_Code != nil {
 					atClaims["specialist_code"] = doctor.Specialist_Code
 					outputData["specialist_code"] = doctor.Specialist_Code