diff --git a/lis2mdl/lis2mdl.go b/lis2mdl/lis2mdl.go
index 04cb822..d1e8611 100644
--- a/lis2mdl/lis2mdl.go
+++ b/lis2mdl/lis2mdl.go
@@ -6,14 +6,15 @@
 package lis2mdl // import "tinygo.org/x/drivers/lis2mdl"
 
 import (
-	"machine"
 	"math"
 	"time"
+
+	"tinygo.org/x/drivers"
 )
 
 // Device wraps an I2C connection to a LIS2MDL device.
 type Device struct {
-	bus        machine.I2C
+	bus        drivers.I2C
 	Address    uint8
 	PowerMode  uint8
 	SystemMode uint8
@@ -31,7 +32,7 @@ type Configuration struct {
 // configured.
 //
 // This function only creates the Device object, it does not touch the device.
-func New(bus machine.I2C) Device {
+func New(bus drivers.I2C) Device {
 	return Device{bus: bus, Address: MAG_ADDRESS}
 }
 
diff --git a/lis2mdl/lis2mdl_test.go b/lis2mdl/lis2mdl_test.go
new file mode 100644
index 0000000..6bf9c45
--- /dev/null
+++ b/lis2mdl/lis2mdl_test.go
@@ -0,0 +1,15 @@
+package lis2mdl
+
+import (
+	"testing"
+
+	qt "github.com/frankban/quicktest"
+	"tinygo.org/x/drivers/tester"
+)
+
+func TestI2CAddress(t *testing.T) {
+	c := qt.New(t)
+	bus := tester.NewI2CBus(c)
+	dev := New(bus)
+	c.Assert(dev.Address, qt.Equals, uint8(MAG_ADDRESS))
+}