diff --git a/compressai_vision/pipelines/base.py b/compressai_vision/pipelines/base.py
index 92e4c1fa..786f4e3d 100755
--- a/compressai_vision/pipelines/base.py
+++ b/compressai_vision/pipelines/base.py
@@ -116,12 +116,14 @@ def acc_kmac_and_pixels_info(self, mname, kmac, pixels):  # for image task
 
     def calc_kmac_per_pixels_image_task(self):  # for video task
         # multiplication
-        self.kmac_per_pixels = {k: (v / self.pixels[k]) for k, v in self.kmacs.items()}
+        self.kmac_per_pixels = {
+            k: (v / self.pixels["nn_part_1"]) for k, v in self.kmacs.items()
+        }
 
     def calc_kmac_per_pixels_video_task(self, nbframes, ori_nbframes):  # for video task
         # multiplication
         self.kmac_per_pixels = {
-            k: (v * nbframes) / (self.pixels[k] * ori_nbframes)
+            k: (v * nbframes) / (self.pixels["nn_part_1"] * ori_nbframes)
             for k, v in self.kmacs.items()
         }