Index: dialin/hd/pretreatment.py
===================================================================
diff -u -r6052189ad455e04a98d23cff2ded8e0fee8b3edf -r688054e96377ce0b2ec5386d7a34b0e1b8a387b3
--- dialin/hd/pretreatment.py	(.../pretreatment.py)	(revision 6052189ad455e04a98d23cff2ded8e0fee8b3edf)
+++ dialin/hd/pretreatment.py	(.../pretreatment.py)	(revision 688054e96377ce0b2ec5386d7a34b0e1b8a387b3)
@@ -21,7 +21,7 @@
                       PreTreatmentConsumableSelfTestStates,
                       PreTreatmentRecircStates,
                       PreTreatmentNoCartSelfTestStates,
-                      PreTreatmentDrySelfTestStates,
+                      PreTreatmentDrySelfTestsStates,
                       PreTreatmentPrimeStates)
 from ..common.msg_defs import MsgIds, MsgFieldPositions
 from ..protocols.CAN import DenaliChannels
@@ -260,12 +260,12 @@
             self.pre_treatment_no_cart_self_test_state = PreTreatmentNoCartSelfTestStates(pt_no_cart_st_state).name
 
         self.pre_treatment_dry_self_test_state = 'Unknown State'
-        if PreTreatmentDrySelfTestStates.has_value(pt_dry_st_state):
-            self.pre_treatment_dry_self_test_state = PreTreatmentDrySelfTestStates(pt_dry_st_state).name
+        if PreTreatmentDrySelfTestsStates.has_value(pt_dry_st_state):
+            self.pre_treatment_dry_self_test_state = PreTreatmentDrySelfTestsStates(pt_dry_st_state).name
 
         self.pre_treatment_dry_self_test_state = 'Unknown State'
-        if PreTreatmentDrySelfTestStates.has_value(pt_dry_st_state):
-            self.pre_treatment_dry_self_test_state = PreTreatmentDrySelfTestStates(pt_dry_st_state).name
+        if PreTreatmentDrySelfTestsStates.has_value(pt_dry_st_state):
+            self.pre_treatment_dry_self_test_state = PreTreatmentDrySelfTestsStates(pt_dry_st_state).name
 
         self.pre_treatment_prime_state = 'Unknown State'
         if PreTreatmentPrimeStates.has_value(pt_prime_state):