Index: tst_TreatmentStatesData/test.py
===================================================================
diff -u -r6d01b12355c38d1a14b9aa6fb3578c52928d285b -r40314c67874695eefc506c3a6a33896495953edd
--- tst_TreatmentStatesData/test.py	(.../test.py)	(revision 6d01b12355c38d1a14b9aa6fb3578c52928d285b)
+++ tst_TreatmentStatesData/test.py	(.../test.py)	(revision 40314c67874695eefc506c3a6a33896495953edd)
@@ -13,8 +13,8 @@
 #
 
 import names
-from dialin.squish import utils, denaliMessages
-from dialin.squish.denaliMessages import txStates
+from dialin.ui import utils
+from dialin.utils import txStates
 
 
 def gotoScreenNtest_Contains_TreatmentStatesData():