diff --git a/Cargo.toml b/Cargo.toml
index d864802..264ef57 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -21,7 +21,7 @@ readme = "README.md"
 [dependencies]
 cortex-m = "0.6"
 cortex-m-rt = "0.6"
-ssd1306 = "0.3"
+ssd1306 = { version = "0.3", path = "../ssd1306" }
 panic-halt = "0.2"
 ina260 = "0.3"
 embedded-graphics = "0.6"
@@ -40,6 +40,12 @@ features = ["cortexm"]
 optional = false
 version = "0.1"
 
+[dependencies.display-interface]
+version = "0.2"
+
+[dependencies.display-interface-i2c]
+version = "0.2"
+
 [profile.dev]
 debug = true
 opt-level = "s"
diff --git a/src/main.rs b/src/main.rs
index 2975b89..74e2e6c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -18,7 +18,7 @@ use embedded_graphics::{
 };
 use ina260::INA260;
 use ssd1306::mode::GraphicsMode;
-use ssd1306::Builder;
+use ssd1306::{Builder, I2CDIBuilder};
 
 use crate::hal::delay::Delay;
 use crate::hal::i2c::*;
@@ -76,7 +76,8 @@ fn main() -> ! {
                 .text_color(BinaryColor::On)
                 .build();
 
-            let mut disp: GraphicsMode<_> = Builder::new().connect_i2c(i2c_bus.acquire()).into();
+            let interface = I2CDIBuilder::new().init(i2c_bus.acquire());
+            let mut disp: GraphicsMode<_> = Builder::new().connect(interface).into();
 
             disp.init().map_err(drop).unwrap();
             disp.flush().map_err(drop).unwrap();