2 files changed, 6 insertions, 6 deletions
diff --git a/Cargo.toml b/Cargo.toml
index d48ac94..daf3a9f 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,14 +1,14 @@
 [package]
 
 name = "yaxpeax-msp430"
-version = "0.0.4"
+version = "0.0.5"
 authors = [ "iximeow <me@iximeow.net>" ]
 license = "0BSD"
 repository = "http://git.iximeow.net/yaxpeax-msp430/"
 description = "msp430 decoders for the yaxpeax project"
 
 [dependencies]
-yaxpeax-arch = { version = "0.0.3", default-features = false, features = [] }
+yaxpeax-arch = { version = "0.0.4", default-features = false, features = [] }
 "termion" = "1.4.0"
 "serde" = { version = "1.0", optional = true }
 "serde_derive" = { version = "1.0", optional = true }
diff --git a/src/lib.rs b/src/lib.rs
index ccaf2ec..7977c88 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -8,7 +8,7 @@ extern crate serde;
 extern crate yaxpeax_arch;
 extern crate termion;
 
-use yaxpeax_arch::{Arch, Decoder, LengthedInstruction};
+use yaxpeax_arch::{Arch, AddressDiff, Decoder, LengthedInstruction};
 
 mod display;
 pub use display::NoContext;
@@ -52,9 +52,9 @@ impl Default for Instruction {
 }
 
 impl LengthedInstruction for Instruction {
-    type Unit = <MSP430 as Arch>::Address;
+    type Unit = AddressDiff<<MSP430 as Arch>::Address>;
     fn min_size() -> Self::Unit {
-        2
+        AddressDiff::from_const(2)
     }
     fn len(&self) -> Self::Unit {
         let mut size = 2;
@@ -72,7 +72,7 @@ impl LengthedInstruction for Instruction {
             Operand::Absolute(_) => { size += 2; },
             _ => {}
         };
-        size
+        AddressDiff::from_const(size)
     }
 }