diff --git a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
index ed94754d6a1bf09459c870dcd2b1cd58661c1a51..729e8bd35671df179525e17d3028ec6a7a5bc05d 100644
--- a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
+++ b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
@@ -142,7 +142,7 @@ def get_loss(model: nn.Module, dataloader: DataLoader):
 
 
 @torch.no_grad()
-def evaluate(model: nn.Module, dataloader: DataLoader = None):
+def evaluate(model: nn.Module, test_dataloader: DataLoader = None):
     model.eval()
     correct = 0
     total = 0