diff --git a/24Hour/Controllers/Common/LawyerArchivesController.cs b/24Hour/Controllers/Common/LawyerArchivesController.cs
index d60fba8..cd34011 100644
--- a/24Hour/Controllers/Common/LawyerArchivesController.cs
+++ b/24Hour/Controllers/Common/LawyerArchivesController.cs
@@ -1646,7 +1646,7 @@ namespace _24Hour.Controllers.Common
                         userim.cardIdphoto = cardids;
                         userim.identityphoto = files;
                         userim.departmentPhoto = departments;
-
+                        userim.unitCode = _userdata.unitCode;
                         _db.BeginTran();
                         var num = await _db.Insertable(userim).ExecuteCommandAsync();
                         _db.CommitTran();
@@ -1656,13 +1656,15 @@ namespace _24Hour.Controllers.Common
                         user.cardIdphoto = cardids;
                         user.identityphoto = files;
                         user.departmentPhoto = departments;
+                        user.unitCode = _userdata.unitCode;
                         _db.BeginTran();
                         var num = await _db.Updateable(user)
                             .UpdateColumns(x => new
                             {
                                 x.cardIdphoto,
                                 x.identityphoto,
-                                x.departmentPhoto
+                                x.departmentPhoto,
+                                x.unitCode,
                             })
                             .IgnoreColumns(ignoreAllNullColumns: true)
                             .ExecuteCommandAsync();