Index: tests/test_training.py
===================================================================
diff -u -r87891e5102d69f7eb8e771f3bb91a9cf72e3b83f -rd10bf8fc145e54979dbeadaedfbaccf0f45d8234
--- tests/test_training.py	(.../test_training.py)	(revision 87891e5102d69f7eb8e771f3bb91a9cf72e3b83f)
+++ tests/test_training.py	(.../test_training.py)	(revision d10bf8fc145e54979dbeadaedfbaccf0f45d8234)
@@ -18,7 +18,7 @@
 sys.path.append("../leahi_dialin")
 from time import sleep
 from leahi_dialin.td.treatment_delivery import TD
-from leahi_dialin.td.modules.constants import RESET, NO_RESET
+from leahi_dialin.common.constants import RESET, NO_RESET
 
 if __name__ == "__main__":
     # create an TD object called td