.

test/test_nwds.py

@@ -1,7 +1,7 @@
 import pytest
-from rota.shifts import NoWorkers, RotaBuilder, SingleShift, days
+from rota_generator.shifts import NoWorkers, RotaBuilder, SingleShift, days
 import datetime
-from rota.workers import Worker
+from rota_generator.workers import Worker
 
 def weeks_from_list(lst):
     """Yield successive n-sized chunks from a lst."""