EXPORT_SYMBOL_GPL(arizona_pm_ops);
 
 #ifdef CONFIG_OF
-unsigned long arizona_of_get_type(struct device *dev)
-{
-       const struct of_device_id *id = of_match_device(arizona_of_match, dev);
-
-       if (id)
-               return (unsigned long)id->data;
-       else
-               return 0;
-}
-EXPORT_SYMBOL_GPL(arizona_of_get_type);
-
 static int arizona_of_get_core_pdata(struct arizona *arizona)
 {
        struct arizona_pdata *pdata = &arizona->pdata;
 
 static int arizona_i2c_probe(struct i2c_client *i2c,
                             const struct i2c_device_id *id)
 {
+       const void *match_data;
        struct arizona *arizona;
        const struct regmap_config *regmap_config = NULL;
-       unsigned long type;
+       unsigned long type = 0;
        int ret;
 
-       if (i2c->dev.of_node)
-               type = arizona_of_get_type(&i2c->dev);
-       else
+       match_data = device_get_match_data(&i2c->dev);
+       if (match_data)
+               type = (unsigned long)match_data;
+       else if (id)
                type = id->driver_data;
 
        switch (type) {
 
 static int arizona_spi_probe(struct spi_device *spi)
 {
        const struct spi_device_id *id = spi_get_device_id(spi);
+       const void *match_data;
        struct arizona *arizona;
        const struct regmap_config *regmap_config = NULL;
-       unsigned long type;
+       unsigned long type = 0;
        int ret;
 
-       if (spi->dev.of_node)
-               type = arizona_of_get_type(&spi->dev);
-       else
+       match_data = device_get_match_data(&spi->dev);
+       if (match_data)
+               type = (unsigned long)match_data;
+       else if (id)
                type = id->driver_data;
 
        switch (type) {
 
 int arizona_irq_init(struct arizona *arizona);
 int arizona_irq_exit(struct arizona *arizona);
 
-#ifdef CONFIG_OF
-unsigned long arizona_of_get_type(struct device *dev);
-#else
-static inline unsigned long arizona_of_get_type(struct device *dev)
-{
-       return 0;
-}
-#endif
-
 #endif