diff --git a/gpio.go b/gpio.go
index 3d7baf7..4f4a091 100644
--- a/gpio.go
+++ b/gpio.go
@@ -151,6 +151,11 @@ type GPIODriver interface {
 var gpioDriverInitialized bool
 var gpioDriverInstance GPIODriver
 
+func SetGpioDriver(drv GPIODriver, initialized bool) {
+	gpioDriverInstance = drv
+	gpioDriverInitialized = initialized
+}
+
 // InitGPIO initializes the GPIO driver.
 func InitGPIO() error {
 	if gpioDriverInitialized {
diff --git a/i2c.go b/i2c.go
index 6a5d1ec..414cd86 100644
--- a/i2c.go
+++ b/i2c.go
@@ -43,6 +43,11 @@ type I2CDriver interface {
 var i2cDriverInitialized bool
 var i2cDriverInstance I2CDriver
 
+func SetI2cDriver(drv I2CDriver, initialized bool) {
+	i2cDriverInstance = drv
+	i2cDriverInitialized = initialized
+}
+
 // InitI2C initializes the I2C driver.
 func InitI2C() error {
 	if i2cDriverInitialized {