diff --git a/base_search_fuzzy/models/trgm_index.py b/base_search_fuzzy/models/trgm_index.py
index f1ea3916f..d4ecc1025 100644
--- a/base_search_fuzzy/models/trgm_index.py
+++ b/base_search_fuzzy/models/trgm_index.py
@@ -2,7 +2,7 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 import logging
 
-from openerp import _, SUPERUSER_ID, api, exceptions, fields, models
+from openerp import SUPERUSER_ID, _, api, exceptions, fields, models
 
 from psycopg2.extensions import AsIs
 
diff --git a/base_search_fuzzy/tests/test_query_generation.py b/base_search_fuzzy/tests/test_query_generation.py
index e094b07e7..34c7e30b3 100644
--- a/base_search_fuzzy/tests/test_query_generation.py
+++ b/base_search_fuzzy/tests/test_query_generation.py
@@ -12,6 +12,7 @@ class QueryGenerationCase(TransactionCase):
         super(QueryGenerationCase, self).setUp()
         self.ResPartner = self.env['res.partner']
         self.TrgmIndex = self.env['trgm.index']
+        self.ResPartnerCategory = self.env['res.partner.category']
 
     def test_fuzzy_where_generation(self):
         """Check the generation of the where clause."""