diff --git a/pvmfw/Android.bp b/pvmfw/Android.bp
index 287e14a..e5fdfe8 100644
--- a/pvmfw/Android.bp
+++ b/pvmfw/Android.bp
@@ -12,7 +12,6 @@
     flags: ["-Dunsafe_op_in_unsafe_fn"],
     features: [
         "legacy",
-        "cpu_feat_hafdbs",
     ],
     rustlibs: [
         "libaarch64_paging",
diff --git a/pvmfw/src/helpers.rs b/pvmfw/src/helpers.rs
index 1d39e1f..dcfd99c 100644
--- a/pvmfw/src/helpers.rs
+++ b/pvmfw/src/helpers.rs
@@ -152,13 +152,3 @@
         core::ffi::CStr::from_bytes_with_nul(concat!($str, "\0").as_bytes()).unwrap()
     }};
 }
-
-/// Returns `true` if hardware dirty state management is available.
-pub fn dbm_available() -> bool {
-    if !cfg!(feature = "cpu_feat_hafdbs") {
-        return false;
-    }
-    // Hardware dirty bit management available flag (ID_AA64MMFR1_EL1.HAFDBS[1])
-    const DBM_AVAILABLE: usize = 1 << 1;
-    read_sysreg!("id_aa64mmfr1_el1") & DBM_AVAILABLE != 0
-}
diff --git a/pvmfw/src/memory.rs b/pvmfw/src/memory.rs
index c316dd2..3205a4d 100644
--- a/pvmfw/src/memory.rs
+++ b/pvmfw/src/memory.rs
@@ -16,7 +16,7 @@
 
 #![deny(unsafe_op_in_unsafe_fn)]
 
-use crate::helpers::{self, dbm_available, page_4kb_of, RangeExt, PVMFW_PAGE_SIZE, SIZE_4MB};
+use crate::helpers::{self, page_4kb_of, RangeExt, PVMFW_PAGE_SIZE, SIZE_4MB};
 use crate::mmu;
 use aarch64_paging::paging::{Attributes, Descriptor, MemoryRegion as VaRange};
 use alloc::alloc::alloc_zeroed;
@@ -40,7 +40,7 @@
 use once_cell::race::OnceBox;
 use spin::mutex::SpinMutex;
 use tinyvec::ArrayVec;
-use vmbase::{dsb, isb, read_sysreg, tlbi, write_sysreg};
+use vmbase::{dsb, isb, memory::set_dbm_enabled, tlbi};
 
 /// Base of the system's contiguous "main" memory.
 pub const BASE_ADDR: usize = 0x8000_0000;
@@ -219,15 +219,13 @@
     const CAPACITY: usize = 5;
     const MMIO_CAPACITY: usize = 5;
     const PVMFW_RANGE: MemoryRange = (BASE_ADDR - SIZE_4MB)..BASE_ADDR;
-    // TCR_EL1.{HA,HD} bits controlling hardware management of access and dirty state
-    const TCR_EL1_HA_HD_BITS: usize = 3 << 39;
 
     /// Create a new instance from an active page table, covering the maximum RAM size.
     pub fn new(mut page_table: mmu::PageTable) -> Self {
         // Activate dirty state management first, otherwise we may get permission faults immediately
         // after activating the new page table. This has no effect before the new page table is
         // activated because none of the entries in the initial idmap have the DBM flag.
-        Self::set_dbm_enabled(true);
+        set_dbm_enabled(true);
 
         debug!("Activating dynamic page table...");
         // SAFETY - page_table duplicates the static mappings for everything that the Rust code is
@@ -434,25 +432,11 @@
             .modify_range(&(addr..addr + 1), &mark_dirty_block)
             .map_err(|_| MemoryTrackerError::SetPteDirtyFailed)
     }
-
-    fn set_dbm_enabled(enabled: bool) {
-        if dbm_available() {
-            let mut tcr = read_sysreg!("tcr_el1");
-            if enabled {
-                tcr |= Self::TCR_EL1_HA_HD_BITS
-            } else {
-                tcr &= !Self::TCR_EL1_HA_HD_BITS
-            };
-            // Safe because it writes to a system register and does not affect Rust.
-            unsafe { write_sysreg!("tcr_el1", tcr) }
-            isb!();
-        }
-    }
 }
 
 impl Drop for MemoryTracker {
     fn drop(&mut self) {
-        Self::set_dbm_enabled(false);
+        set_dbm_enabled(false);
         self.flush_dirty_pages().unwrap();
         self.unshare_all_memory();
     }