diff --git a/src/omniperf_soc/soc_base.py b/src/omniperf_soc/soc_base.py
index d0b01cbfda..76e798439b 100644
--- a/src/omniperf_soc/soc_base.py
+++ b/src/omniperf_soc/soc_base.py
@@ -195,10 +195,7 @@ class OmniSoC_Base:
             ):
                 self._mspec.gpu_model = "MI300X_A1"
             # We need to distinguish MI308X by peeking reported num CUs
-            elif(
-                self._mspec.cu_per_gpu == "80"
-                or "MI308X" in self.check_arch_override()
-            ):
+            elif self._mspec.cu_per_gpu == "80" or "MI308X" in self.check_arch_override():
                 self._mspec.gpu_model = "MI308X"
             else:
                 console_error(
diff --git a/src/utils/utils.py b/src/utils/utils.py
index 9869c562a0..b99d6a0b43 100644
--- a/src/utils/utils.py
+++ b/src/utils/utils.py
@@ -562,7 +562,10 @@ def total_xcds(archname, compute_partition):
     mi300a_archs = ["mi300a_a0", "mi300a_a1"]
     mi300x_archs = ["mi300x_a0", "mi300x_a1"]
     mi308x_archs = ["mi308x"]
-    if archname.lower() in mi300a_archs + mi300x_archs + mi308x_archs and compute_partition == "NA":
+    if (
+        archname.lower() in mi300a_archs + mi300x_archs + mi308x_archs
+        and compute_partition == "NA"
+    ):
         console_error("Invalid compute partition found for {}".format(archname))
     if archname.lower() not in mi300a_archs + mi300x_archs + mi308x_archs:
         return 1