diff --git a/modules/worker/back/methods/worker-time-control/specs/login.spec.js b/modules/worker/back/methods/worker-time-control/specs/login.spec.js
index e125a876d..eff0edaa7 100644
--- a/modules/worker/back/methods/worker-time-control/specs/login.spec.js
+++ b/modules/worker/back/methods/worker-time-control/specs/login.spec.js
@@ -1,5 +1,4 @@
 const models = require('vn-loopback/server/server').models;
-const LoopBackContext = require('loopback-context');
 const UserError = require('vn-loopback/util/user-error');
 
 describe('workerTimeControl login()', () => {
diff --git a/modules/worker/back/methods/worker-time-control/specs/sendMail.spec.js b/modules/worker/back/methods/worker-time-control/specs/sendMail.spec.js
index 133d7271c..1395c7a46 100644
--- a/modules/worker/back/methods/worker-time-control/specs/sendMail.spec.js
+++ b/modules/worker/back/methods/worker-time-control/specs/sendMail.spec.js
@@ -2,17 +2,24 @@ const models = require('vn-loopback/server/server').models;
 
 describe('sendMail', () => {
     it('should insert in mail', async() => {
+        const tx = await models.Sale.beginTransaction({});
+        const options = {transaction: tx};
+        options.transaction = tx;
+        let mailCountBefore;
+        let mailCountAfter;
         const ctx = {
             req: {accessToken: {userId: 50}},
             args: {workerFk: 1106, year: 2001, week: 1}
         };
-        const tx = await models.Sale.beginTransaction({});
-        const options = {transaction: tx};
-        options.transaction = tx;
 
-        const mailCountBefore = await models.Mail.count(options);
-        await models.WorkerTimeControl.sendMail(ctx, options);
-        const mailCountAfter = await models.Mail.count(options);
+        try {
+            mailCountBefore = await models.Mail.count(options);
+            await models.WorkerTimeControl.sendMail(ctx, options);
+            mailCountAfter = await models.Mail.count(options);
+        } catch (e) {
+            await tx.rollback();
+            throw e;
+        }
 
         expect(mailCountAfter).toBeGreaterThan(mailCountBefore);
         await tx.rollback();