struct ingress_sched_data {
        struct tcf_block *block;
+       struct tcf_block_ext_info block_info;
 };
 
 static struct Qdisc *ingress_leaf(struct Qdisc *sch, unsigned long arg)
        struct net_device *dev = qdisc_dev(sch);
        int err;
 
-       err = tcf_block_get(&q->block, &dev->ingress_cl_list, sch);
+       q->block_info.binder_type = TCF_BLOCK_BINDER_TYPE_CLSACT_INGRESS;
+
+       err = tcf_block_get_ext(&q->block, &dev->ingress_cl_list,
+                               sch, &q->block_info);
        if (err)
                return err;
 
 static void ingress_destroy(struct Qdisc *sch)
 {
        struct ingress_sched_data *q = qdisc_priv(sch);
+       struct net_device *dev = qdisc_dev(sch);
 
-       tcf_block_put(q->block);
+       tcf_block_put_ext(q->block, &dev->ingress_cl_list,
+                         sch, &q->block_info);
        net_dec_ingress_queue();
 }
 
 struct clsact_sched_data {
        struct tcf_block *ingress_block;
        struct tcf_block *egress_block;
+       struct tcf_block_ext_info ingress_block_info;
+       struct tcf_block_ext_info egress_block_info;
 };
 
 static unsigned long clsact_find(struct Qdisc *sch, u32 classid)
        struct net_device *dev = qdisc_dev(sch);
        int err;
 
-       err = tcf_block_get(&q->ingress_block, &dev->ingress_cl_list, sch);
+       q->ingress_block_info.binder_type = TCF_BLOCK_BINDER_TYPE_CLSACT_INGRESS;
+
+       err = tcf_block_get_ext(&q->ingress_block, &dev->ingress_cl_list,
+                               sch, &q->ingress_block_info);
        if (err)
                return err;
 
-       err = tcf_block_get(&q->egress_block, &dev->egress_cl_list, sch);
+       q->egress_block_info.binder_type = TCF_BLOCK_BINDER_TYPE_CLSACT_EGRESS;
+
+       err = tcf_block_get_ext(&q->egress_block, &dev->egress_cl_list,
+                               sch, &q->egress_block_info);
        if (err)
-               return err;
+               goto err_egress_block_get;
 
        net_inc_ingress_queue();
        net_inc_egress_queue();
        sch->flags |= TCQ_F_CPUSTATS;
 
        return 0;
+
+err_egress_block_get:
+       tcf_block_put_ext(q->ingress_block, &dev->ingress_cl_list,
+                         sch, &q->ingress_block_info);
+       return err;
 }
 
 static void clsact_destroy(struct Qdisc *sch)
 {
        struct clsact_sched_data *q = qdisc_priv(sch);
+       struct net_device *dev = qdisc_dev(sch);
 
-       tcf_block_put(q->egress_block);
-       tcf_block_put(q->ingress_block);
+       tcf_block_put_ext(q->egress_block, &dev->egress_cl_list,
+                         sch, &q->egress_block_info);
+       tcf_block_put_ext(q->ingress_block, &dev->ingress_cl_list,
+                         sch, &q->ingress_block_info);
 
        net_dec_ingress_queue();
        net_dec_egress_queue();