diff --git a/tests/test_frab_export.py b/tests/test_frab_export.py
index 640b31c84..382b0b4ed 100644
--- a/tests/test_frab_export.py
+++ b/tests/test_frab_export.py
@@ -3,7 +3,7 @@
 from lxml import etree
 
 from apps.schedule import event_tz
-from apps.schedule.schedule_xml import (
+from apps.schedule.schedule_frab_xml import (
     make_root,
     add_day,
     add_room,
diff --git a/tests/test_schedule.py b/tests/test_schedule.py
index 18550113b..5a0118ce7 100644
--- a/tests/test_schedule.py
+++ b/tests/test_schedule.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from apps.schedule import schedule_xml
+from apps.schedule import schedule_frab_xml
 
 
 @pytest.mark.parametrize('start_time, end_time, expected', [
@@ -17,4 +17,4 @@ def test_get_duration(start_time, end_time, expected):
     fmt = '%Y-%m-%d %H:%M:%S'
     start_time = datetime.strptime(start_time, fmt)
     end_time = datetime.strptime(end_time, fmt)
-    assert schedule_xml.get_duration(start_time, end_time) == expected
+    assert schedule_frab_xml.get_duration(start_time, end_time) == expected