diff --git a/company_today/tests/test_today.py b/company_today/tests/test_today.py
index 406e6ee71..a2feffcc9 100644
--- a/company_today/tests/test_today.py
+++ b/company_today/tests/test_today.py
@@ -12,7 +12,7 @@ class TestToday(TransactionCase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()
-        cls.company = cls.ref("base.main_company")
+        cls.company = cls.env.ref("base.main_company")
 
     def test_today(self):
         self.company.cron_update_today()