#include "tlv320aic32x4.h"
 
-static const struct of_device_id aic32x4_of_id[];
-
 static int aic32x4_spi_probe(struct spi_device *spi)
 {
        struct regmap *regmap;
        config.read_flag_mask = 0x01;
 
        regmap = devm_regmap_init_spi(spi, &config);
-
-       if (spi->dev.of_node) {
-               const struct of_device_id *oid;
-
-               oid = of_match_node(aic32x4_of_id, spi->dev.of_node);
-               type = (uintptr_t)oid->data;
-       } else {
-               const struct spi_device_id *id_entry;
-
-               id_entry = spi_get_device_id(spi);
-               type = id_entry->driver_data;
-       }
+       type = (uintptr_t)spi_get_device_match_data(spi);
 
        return aic32x4_probe(&spi->dev, regmap, type);
 }