diff --git a/models/account_move.py b/models/account_move.py
index 82efbed..baf1a87 100644
--- a/models/account_move.py
+++ b/models/account_move.py
@@ -22,6 +22,6 @@ class AccountMove(models.Model):
     def write(self, vals):
         # Surgical Jumper to bypass hr_expense's account.move lock
         hr_expense_class = self._get_hr_expense_base_class()
-        if hr_expense_class and self._context.get('skip_expense_lock'):
+        if hr_expense_class and self.env.context.get('skip_expense_lock'):
             return super(hr_expense_class, self).write(vals)
         return super().write(vals)
diff --git a/models/account_move_line.py b/models/account_move_line.py
index 9af8423..f2c682a 100644
--- a/models/account_move_line.py
+++ b/models/account_move_line.py
@@ -28,7 +28,7 @@ class AccountMoveLine(models.Model):
         
         # 2. Surgical Jumper to bypass hr_expense's account.move.line lock
         hr_expense_class = self._get_hr_expense_base_class()
-        if hr_expense_class and self._context.get('skip_expense_lock'):
+        if hr_expense_class and self.env.context.get('skip_expense_lock'):
             return super(hr_expense_class, self).write(vals)
 
         return super().write(vals)
@@ -36,6 +36,6 @@ class AccountMoveLine(models.Model):
     def unlink(self):
         # Surgical Jumper to bypass hr_expense's account.move.line lock
         hr_expense_class = self._get_hr_expense_base_class()
-        if hr_expense_class and self._context.get('skip_expense_lock'):
+        if hr_expense_class and self.env.context.get('skip_expense_lock'):
             return super(hr_expense_class, self).unlink()
         return super().unlink()