diff --git a/tests/test_ics_writer.py b/tests/test_ics_writer.py
index 9e5d3f7..c707997 100644
--- a/tests/test_ics_writer.py
+++ b/tests/test_ics_writer.py
@@ -2,7 +2,8 @@
 from ics import Calendar
 from freezegun import freeze_time
 
-from fb2cal import ICSWriter, FacebookUser
+from fb2cal.ics_writer import ICSWriter
+from fb2cal.facebook_user import FacebookUser
 
 class TestICSWriter(unittest.TestCase):
     def setUp(self):
diff --git a/tests/test_transformer.py b/tests/test_transformer.py
index 712530d..7412ef5 100644
--- a/tests/test_transformer.py
+++ b/tests/test_transformer.py
@@ -1,5 +1,5 @@
 import unittest
-from fb2cal import Transformer
+from fb2cal.transformer import Transformer
 
 from mocks.birthday_comet_root_mocks import BIRTHDAY_COMET_ROOT_JANUARY_MOCK