diff --git a/src/omniperf_soc/soc_base.py b/src/omniperf_soc/soc_base.py
index 8a48087269..db8ce9a224 100644
--- a/src/omniperf_soc/soc_base.py
+++ b/src/omniperf_soc/soc_base.py
@@ -187,7 +187,7 @@ class OmniSoC_Base:
                 self._mspec.chip_id = MI300_CHIP_IDS[self._mspec.chip_id]
 
         self._mspec.num_xcd = str(
-            total_xcds(self._mspec.gpu_model, self._mspec.compute_partition)
+            total_xcds(self._mspec.chip_id, self._mspec.compute_partition)
         )
 
     @demarcate