#include <linux/iio/iio.h>
 #include <linux/iio/types.h>
 #include <linux/module.h>
+#include <linux/mod_devicetable.h>
 #include <linux/mutex.h>
-#include <linux/of.h>
-#include <linux/of_device.h>
+#include <linux/property.h>
 #include <linux/spi/spi.h>
 
 #define MCP4131_WRITE          (0x00 << 2)
        data = iio_priv(indio_dev);
        spi_set_drvdata(spi, indio_dev);
        data->spi = spi;
-       data->cfg = of_device_get_match_data(&spi->dev);
+       data->cfg = device_get_match_data(&spi->dev);
        if (!data->cfg) {
                devid = spi_get_device_id(spi)->driver_data;
                data->cfg = &mcp4131_cfg[devid];
 static struct spi_driver mcp4131_driver = {
        .driver = {
                .name   = "mcp4131",
-               .of_match_table = of_match_ptr(mcp4131_dt_ids),
+               .of_match_table = mcp4131_dt_ids,
        },
        .probe          = mcp4131_probe,
        .id_table       = mcp4131_id,