Index: tst_TreatmentStatesData/test.py
===================================================================
diff -u -r6d01b12355c38d1a14b9aa6fb3578c52928d285b -r316fd2f51b0fa89c0f69949d79ef246e1e4bc4ed
--- tst_TreatmentStatesData/test.py	(.../test.py)	(revision 6d01b12355c38d1a14b9aa6fb3578c52928d285b)
+++ tst_TreatmentStatesData/test.py	(.../test.py)	(revision 316fd2f51b0fa89c0f69949d79ef246e1e4bc4ed)
@@ -1,20 +1,23 @@
 # -*- coding: utf-8 -*-
-
-##
-# Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
-# copyright
-#          THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM,
-#          IN PART OR IN WHOLE,
-#          WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
+###########################################################################
 #
-# file    tst_TreatmentStatesData
-# date    2020/07/26
-# author  Behrouz NematiPour
+# Copyright (c) 2020-2025 Diality Inc. - All Rights Reserved.
 #
+# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
+# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
+#
+# @file    test.py
+#
+# @author  (last)      Vy
+# @date    (last)      01-Dec-2023
+# @author  (original)  Behrouz NemaiPour
+# @date    (original)  28-Jul-2020
+#
+############################################################################
 
 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():