diff --git a/drivers/misc/pmic_i2c.c b/drivers/misc/pmic_i2c.c
index b82e899..ad55d64 100644
--- a/drivers/misc/pmic_i2c.c
+++ b/drivers/misc/pmic_i2c.c
@@ -81,7 +81,7 @@
 
 int pmic_probe(struct pmic *p)
 {
-	i2c_set_bus_num(p->bus);
+	I2C_SET_BUS(p->bus);
 	debug("PMIC:%s probed!\n", p->name);
 	if (i2c_probe(pmic_i2c_addr)) {
 		printf("Can't find PMIC:%s\n", p->name);