static int __bpf_offload_ndo(struct bpf_prog *prog, enum bpf_netdev_command cmd,
                             struct netdev_bpf *data)
 {
-       struct net_device *netdev = prog->aux->offload->netdev;
+       struct bpf_dev_offload *offload = prog->aux->offload;
+       struct net_device *netdev;
 
        ASSERT_RTNL();
 
-       if (!netdev)
+       if (!offload)
                return -ENODEV;
+       netdev = offload->netdev;
        if (!netdev->netdev_ops->ndo_bpf)
                return -EOPNOTSUPP;
 
 
        down_read(&bpf_devs_lock);
        offload = env->prog->aux->offload;
-       if (offload->netdev)
+       if (offload)
                ret = offload->dev_ops->insn_hook(env, insn_idx, prev_insn_idx);
        up_read(&bpf_devs_lock);
 
        struct bpf_dev_offload *offload = prog->aux->offload;
        struct netdev_bpf data = {};
 
-       /* Caution - if netdev is destroyed before the program, this function
-        * will be called twice.
-        */
-
        data.offload.prog = prog;
 
        if (offload->dev_state)
                WARN_ON(__bpf_offload_ndo(prog, BPF_OFFLOAD_DESTROY, &data));
 
-       offload->dev_state = false;
        list_del_init(&offload->offloads);
-       offload->netdev = NULL;
+       kfree(offload);
+       prog->aux->offload = NULL;
 }
 
 void bpf_prog_offload_destroy(struct bpf_prog *prog)
 {
-       struct bpf_dev_offload *offload = prog->aux->offload;
-
        rtnl_lock();
        down_write(&bpf_devs_lock);
-       __bpf_prog_offload_destroy(prog);
+       if (prog->aux->offload)
+               __bpf_prog_offload_destroy(prog);
        up_write(&bpf_devs_lock);
        rtnl_unlock();
-
-       kfree(offload);
 }
 
 static int bpf_prog_offload_translate(struct bpf_prog *prog)