if (err)
                goto out_put_ports;
 
-       devlink_register(devlink);
        err = mscc_ocelot_init_ports(pdev, ports);
        if (err)
                goto out_ocelot_devlink_unregister;
        register_switchdev_blocking_notifier(&ocelot_switchdev_blocking_nb);
 
        of_node_put(ports);
+       devlink_register(devlink);
 
        dev_info(&pdev->dev, "Ocelot switch probed\n");
 
        mscc_ocelot_release_ports(ocelot);
        mscc_ocelot_teardown_devlink_ports(ocelot);
 out_ocelot_devlink_unregister:
-       devlink_unregister(devlink);
        ocelot_deinit(ocelot);
 out_put_ports:
        of_node_put(ports);
 {
        struct ocelot *ocelot = platform_get_drvdata(pdev);
 
+       devlink_unregister(ocelot->devlink);
        ocelot_deinit_timestamp(ocelot);
        ocelot_devlink_sb_unregister(ocelot);
        mscc_ocelot_release_ports(ocelot);
        mscc_ocelot_teardown_devlink_ports(ocelot);
-       devlink_unregister(ocelot->devlink);
        ocelot_deinit(ocelot);
        unregister_switchdev_blocking_notifier(&ocelot_switchdev_blocking_nb);
        unregister_switchdev_notifier(&ocelot_switchdev_nb);