diff --git a/src/python/gem5/components/cachehierarchies/abstract_cache_hierarchy.py b/src/python/gem5/components/cachehierarchies/abstract_cache_hierarchy.py
index 13b4d4f2a2..b0435543af 100644
--- a/src/python/gem5/components/cachehierarchies/abstract_cache_hierarchy.py
+++ b/src/python/gem5/components/cachehierarchies/abstract_cache_hierarchy.py
@@ -53,7 +53,7 @@ class CacheNode:
         self,
         name: str,
         cache: SimObject,
-        next_level: int,
+        next_level: "CacheNode",
         hierarchy: "AbstractCacheHierarchy",
     ):
         self.name = name