diff --git a/hpvm/test/dnn_benchmarks/benchmarks/alexnet/alexnet.cpp b/hpvm/test/dnn_benchmarks/benchmarks/alexnet/alexnet.cpp
index 8202cbccfeefdde16e466c7d741e7f63a86b192a..4dcd57c8164c8bd73280d6224c44bb8b9ec9d6f0 100644
--- a/hpvm/test/dnn_benchmarks/benchmarks/alexnet/alexnet.cpp
+++ b/hpvm/test/dnn_benchmarks/benchmarks/alexnet/alexnet.cpp
@@ -370,7 +370,7 @@ int main() {
   std::string input_path = dir_prefix + std::string("input.bin");
   void *input = readTrainedWeights(input_path.c_str(), 0, 5000, 3, 32, 32);
   std::string labels_path = dir_prefix + std::string("labels.bin");
-  uint8_t *labels = readLabels(labels_path.c_str(), 5000);
+  uint32_t *labels = readLabels3(labels_path.c_str(), 5000);
   std::string conv2d_1_w_path = dir_prefix + std::string("conv2d_1_w.bin");
   void *conv2d_1_w =
       readTrainedWeights(conv2d_1_w_path.c_str(), 0, 64, 3, 11, 11);
@@ -444,6 +444,6 @@ int main() {
   hpvm_request_tensor(result, 0);
 
   __hpvm__cleanup();
-  computeAccuracy2(labels, 5000, result);
+  computeAccuracy3(labels, result);
   return 0;
 }
diff --git a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp
index f91e148d95e5f238110e9ffc53ce37b392af0690..b32dccabc2f29b54e8da35551f8d982cd13a378c 100644
--- a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp
+++ b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp
@@ -2503,7 +2503,7 @@ int main() {
   std::string dense_1_b_path = dir_prefix + std::string("dense_1_b.bin");
   void *dense_1_b = readTrainedWeights(dense_1_b_path.c_str(), 0, 1, 10, 1, 1);
   void *input = readTrainedWeights(input_path.c_str(), 0, 5000, 3, 32, 32);
-  uint8_t *labels = readLabels(labels_path.c_str(), 5000);
+  uint32_t *labels = readLabels3(labels_path.c_str(), 5000);
 
   __hpvm__init();
   RootIn *args = static_cast<RootIn *>(malloc(sizeof(RootIn)));
@@ -2793,6 +2793,6 @@ int main() {
   hpvm_request_tensor(result, 0);
 
   __hpvm__cleanup();
-  computeAccuracy2(labels, 5000, result);
+  computeAccuracy3(labels, result);
   return 0;
 }