struct wm8962_priv *wm8962 = snd_soc_codec_get_drvdata(codec);
        int ret;
 
-       wm8962->beep = input_allocate_device();
+       wm8962->beep = devm_input_allocate_device(codec->dev);
        if (!wm8962->beep) {
                dev_err(codec->dev, "Failed to allocate beep device\n");
                return;
 
        ret = input_register_device(wm8962->beep);
        if (ret != 0) {
-               input_free_device(wm8962->beep);
                wm8962->beep = NULL;
                dev_err(codec->dev, "Failed to register beep device\n");
        }
        struct wm8962_priv *wm8962 = snd_soc_codec_get_drvdata(codec);
 
        device_remove_file(codec->dev, &dev_attr_beep);
-       input_unregister_device(wm8962->beep);
        cancel_work_sync(&wm8962->beep_work);
        wm8962->beep = NULL;