diff --git a/tests.py b/tests.py
index 1957465..e401321 100644
--- a/tests.py
+++ b/tests.py
@@ -6,10 +6,10 @@ All rights reserved.
 
 License: GNU General Public License
 '''
-import update.version
+import gitea_auto_update.lib.version
 import unittest
 
-version = update.version.Version('', '')
+version = gitea_auto_update.lib.version.Version('', '')
 
 class Tests(unittest.TestCase):