1 files changed, 4 insertions, 4 deletions
diff --git a/src/lib.rs b/src/lib.rs
index e83a8b0..074d660 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -7,7 +7,7 @@ use std::mem;
 
 use num_enum::IntoPrimitive;
 
-use yaxpeax_arch::{Arch, Decoder, LengthedInstruction};
+use yaxpeax_arch::{Arch, AddressDiff, Decoder, LengthedInstruction};
 
 mod display;
 
@@ -88,13 +88,13 @@ impl yaxpeax_arch::Instruction for Instruction {
 }
 
 impl LengthedInstruction for Instruction {
-    type Unit = u32;
+    type Unit = AddressDiff<u32>;
     fn min_size() -> Self::Unit {
-        4
+        AddressDiff::from_const(4)
     }
 
     fn len(&self) -> Self::Unit {
-        4
+        AddressDiff::from_const(4)
     }
 }