diff --git a/beesdoo_shift/models/task.py b/beesdoo_shift/models/task.py
index 4e0331d..587bfa2 100644
--- a/beesdoo_shift/models/task.py
+++ b/beesdoo_shift/models/task.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from openerp import models, fields, api
+from openerp import models, fields, api, _
 from openerp.exceptions import UserError
 
 class TaskStage(models.Model):