summaryrefslogtreecommitdiff
path: root/rust/src
diff options
context:
space:
mode:
Diffstat (limited to 'rust/src')
-rw-r--r--rust/src/architecture.rs3
1 files changed, 2 insertions, 1 deletions
diff --git a/rust/src/architecture.rs b/rust/src/architecture.rs
index 6df8b19c..7ce672b8 100644
--- a/rust/src/architecture.rs
+++ b/rust/src/architecture.rs
@@ -1729,7 +1729,8 @@ where
let custom_arch = unsafe { &*(ctxt as *mut A) };
custom_arch
.flag_write_from_id(write_type)
- .map(|w| w.id())
+ .map(|w| w.class())
+ .and_then(|c| c.map(|c| c.id()))
.unwrap_or(0)
}