diff --git a/src/python/gem5/components/processors/cpu_types.py b/src/python/gem5/components/processors/cpu_types.py
index e5cc61eead..e12eb99816 100644
--- a/src/python/gem5/components/processors/cpu_types.py
+++ b/src/python/gem5/components/processors/cpu_types.py
@@ -39,7 +39,7 @@ class CPUTypes(Enum):
     MINOR = "minor"
 
 
-def get_cpu_types_str_set() -> Set[CPUTypes]:
+def get_cpu_types_str_set() -> Set[str]:
     """
     Returns a set of all the CPU types as strings.
     """
diff --git a/src/python/gem5/isas.py b/src/python/gem5/isas.py
index 2fb35eb81b..c904c9d227 100644
--- a/src/python/gem5/isas.py
+++ b/src/python/gem5/isas.py
@@ -57,7 +57,7 @@ class ISA(Enum):
     NULL = "null"
 
 
-def get_isas_str_set() -> Set[ISA]:
+def get_isas_str_set() -> Set[str]:
     """
     Returns a set of all the ISA as strings.
     """