diff --git a/codes/autotest/do_current.py b/codes/autotest/do_current.py
index a9c9ea2..9137feb 100644
--- a/codes/autotest/do_current.py
+++ b/codes/autotest/do_current.py
@@ -111,9 +111,9 @@ class DoCurrentTest(QThread):
                         d_sensor[axis].extend(d_item)
                     elif item.get("channel", None) == axis and item.get("name", None) == "hw_estimate_trans_trq_res":
                         d_trans[axis].extend(d_item)
-                    elif item.get("channel", None) == number and item.get("name", None) == "hw_predict_trq_res":
+                    elif item.get("channel", None) == axis and item.get("name", None) == "hw_predict_trq_res":
                         d_predict_trq[axis].extend(d_item)
-                    elif item.get("channel", None) == number and item.get("name", None) == "hw_real_trq_res":
+                    elif item.get("channel", None) == axis and item.get("name", None) == "hw_real_trq_res":
                         d_real_trq[axis].extend(d_item)
 
         for axis in range(6):