#define MAX_LANES                      2
 
 struct j721e_pcie {
-       struct device           *dev;
+       struct cdns_pcie        *cdns_pcie;
        struct clk              *refclk;
        u32                     mode;
        u32                     num_lanes;
-       struct cdns_pcie        *cdns_pcie;
        void __iomem            *user_cfg_base;
        void __iomem            *intd_cfg_base;
        u32                     linkdown_irq_regfield;
 static irqreturn_t j721e_pcie_link_irq_handler(int irq, void *priv)
 {
        struct j721e_pcie *pcie = priv;
-       struct device *dev = pcie->dev;
+       struct device *dev = pcie->cdns_pcie->dev;
        u32 reg;
 
        reg = j721e_pcie_intd_readl(pcie, STATUS_REG_SYS_2);
 static int j721e_pcie_set_mode(struct j721e_pcie *pcie, struct regmap *syscon,
                               unsigned int offset)
 {
-       struct device *dev = pcie->dev;
+       struct device *dev = pcie->cdns_pcie->dev;
        u32 mask = J721E_MODE_RC;
        u32 mode = pcie->mode;
        u32 val = 0;
 static int j721e_pcie_set_link_speed(struct j721e_pcie *pcie,
                                     struct regmap *syscon, unsigned int offset)
 {
-       struct device *dev = pcie->dev;
+       struct device *dev = pcie->cdns_pcie->dev;
        struct device_node *np = dev->of_node;
        int link_speed;
        u32 val = 0;
 static int j721e_pcie_set_lane_count(struct j721e_pcie *pcie,
                                     struct regmap *syscon, unsigned int offset)
 {
-       struct device *dev = pcie->dev;
+       struct device *dev = pcie->cdns_pcie->dev;
        u32 lanes = pcie->num_lanes;
        u32 val = 0;
        int ret;
 
 static int j721e_pcie_ctrl_init(struct j721e_pcie *pcie)
 {
-       struct device *dev = pcie->dev;
+       struct device *dev = pcie->cdns_pcie->dev;
        struct device_node *node = dev->of_node;
        struct of_phandle_args args;
        unsigned int offset = 0;
        if (!pcie)
                return -ENOMEM;
 
-       pcie->dev = dev;
        pcie->mode = mode;
        pcie->linkdown_irq_regfield = data->linkdown_irq_regfield;