#include "ucsi.h"
 #include "trace.h"
 
-#define to_ucsi_connector(_cap_) container_of(_cap_, struct ucsi_connector, \
-                                             typec_cap)
-
 /*
  * UCSI_TIMEOUT_MS - PPM communication timeout
  *
        return ret;
 }
 
-static int
-ucsi_dr_swap(const struct typec_capability *cap, enum typec_data_role role)
+static int ucsi_dr_swap(struct typec_port *port, enum typec_data_role role)
 {
-       struct ucsi_connector *con = to_ucsi_connector(cap);
+       struct ucsi_connector *con = typec_get_drvdata(port);
        struct ucsi_control ctrl;
        int ret = 0;
 
        return ret < 0 ? ret : 0;
 }
 
-static int
-ucsi_pr_swap(const struct typec_capability *cap, enum typec_role role)
+static int ucsi_pr_swap(struct typec_port *port, enum typec_role role)
 {
-       struct ucsi_connector *con = to_ucsi_connector(cap);
+       struct ucsi_connector *con = typec_get_drvdata(port);
        struct ucsi_control ctrl;
        int ret = 0;
 
        return ret;
 }
 
+static const struct typec_operations ucsi_ops = {
+       .dr_set = ucsi_dr_swap,
+       .pr_set = ucsi_pr_swap
+};
+
 static struct fwnode_handle *ucsi_find_fwnode(struct ucsi_connector *con)
 {
        struct fwnode_handle *fwnode;
                *accessory = TYPEC_ACCESSORY_DEBUG;
 
        cap->fwnode = ucsi_find_fwnode(con);
-       cap->dr_set = ucsi_dr_swap;
-       cap->pr_set = ucsi_pr_swap;
+       cap->driver_data = con;
+       cap->ops = &ucsi_ops;
 
        /* Register the connector */
        con->port = typec_register_port(ucsi->dev, cap);