diff --git a/src/arch/arm/table_walker.cc b/src/arch/arm/table_walker.cc
index 1ce2f65800..6f4116dbe0 100644
--- a/src/arch/arm/table_walker.cc
+++ b/src/arch/arm/table_walker.cc
@@ -843,6 +843,7 @@ TableWalker::processWalkLPAE()
     currState->longDesc.lookupLevel = start_lookup_level;
     currState->longDesc.aarch64 = false;
     currState->longDesc.grainSize = Grain4KB;
+    currState->longDesc.isStage2 = isStage2;
 
     fetchDescriptor(
         desc_addr, currState->longDesc,
@@ -1095,6 +1096,7 @@ TableWalker::processWalkAArch64()
     currState->longDesc.aarch64 = true;
     currState->longDesc.grainSize = tg;
     currState->longDesc.physAddrRange = _physAddrRange;
+    currState->longDesc.isStage2 = isStage2;
 
     fetchDescriptor(desc_addr, currState->longDesc,
                     sizeof(uint64_t), flag, start_lookup_level,
diff --git a/src/arch/arm/table_walker.hh b/src/arch/arm/table_walker.hh
index 5143913f4f..66d5ea3674 100644
--- a/src/arch/arm/table_walker.hh
+++ b/src/arch/arm/table_walker.hh
@@ -436,7 +436,7 @@ class TableWalker : public ClockedObject
 
         LongDescriptor()
           : data(0), _dirty(false), aarch64(false), grainSize(Grain4KB),
-            physAddrRange(0)
+            physAddrRange(0), isStage2(false)
         {}
 
         /** The raw bits of the entry */
@@ -454,6 +454,8 @@ class TableWalker : public ClockedObject
 
         uint8_t physAddrRange;
 
+        bool isStage2;
+
         uint8_t*
         getRawPtr() override
         {