diff --git a/tests/test_generic.py b/tests/test_generic.py
index bbcb1b74..1cd21fbc 100644
--- a/tests/test_generic.py
+++ b/tests/test_generic.py
@@ -14,9 +14,10 @@ import unittest
 import xbmc
 
 if 'KOD_TST_CH' not in os.environ:
+    from sakee import addoninfo
     # custom paths
     def add_on_info(*args, **kwargs):
-        return xbmc.AddonData(
+        return addoninfo.AddonData(
             kodi_home_path=os.path.join(os.getcwd(), 'tests', 'home'),
             add_on_id='plugin.video.kod',
             add_on_path=os.getcwd(),
@@ -24,7 +25,6 @@ if 'KOD_TST_CH' not in os.environ:
         )
 
     # override
-    from sakee import addoninfo
     addoninfo.get_add_on_info_from_calling_script = add_on_info
 
 # functions that on kodi 19 moved to xbmcvfs