diff --git a/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py b/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py
index ba9379200..8579fc1e5 100644
--- a/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py
+++ b/src/vai_optimizer/pytorch_binding/pytorch_nndct/qproc/adaquant.py
@@ -882,7 +882,7 @@ def optimize_layer_v2(self, qnode, float_layer, layer_act_pair, net_inputs, net_
     total_loss = AverageMeter("layer_loss")
     best_params = self.get_layer_params(layer)
     # torch version >= 1.6
-    if compare_torch_version('1.6.0',CmpFlag.GREATER_EQUAL):
+    if compare_torch_version(CmpFlag.GREATER_EQUAL, '1.6.0'):
       act_func_map = {
         NNDCT_OP.RELU: F.relu,
         NNDCT_OP.RELU6: F.relu6,
diff --git a/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py b/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py
index ba9379200..8579fc1e5 100644
--- a/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py
+++ b/src/vai_quantizer/vai_q_pytorch/pytorch_binding/pytorch_nndct/qproc/adaquant.py
@@ -882,7 +882,7 @@ def optimize_layer_v2(self, qnode, float_layer, layer_act_pair, net_inputs, net_
     total_loss = AverageMeter("layer_loss")
     best_params = self.get_layer_params(layer)
     # torch version >= 1.6
-    if compare_torch_version('1.6.0',CmpFlag.GREATER_EQUAL):
+    if compare_torch_version(CmpFlag.GREATER_EQUAL, '1.6.0'):
       act_func_map = {
         NNDCT_OP.RELU: F.relu,
         NNDCT_OP.RELU6: F.relu6,