diff --git a/tests/serialization_test.py b/tests/serialization_test.py
index 8349505..8682208 100644
--- a/tests/serialization_test.py
+++ b/tests/serialization_test.py
@@ -1,6 +1,6 @@
 import unittest
 
-from regexsolver.regexsolver import GenerateStringsRequest, MultiTermsRequest, Term
+from regexsolver import GenerateStringsRequest, MultiTermsRequest, Term
 
 
 class SerializationTest(unittest.TestCase):