static int avs_nau8825_codec_init(struct snd_soc_pcm_runtime *runtime)
 {
-       struct snd_soc_dai *codec_dai = asoc_rtd_to_codec(runtime, 0);
-       struct snd_soc_component *component = codec_dai->component;
+       struct snd_soc_card *card = runtime->card;
        struct snd_soc_jack_pin *pins;
        struct snd_soc_jack *jack;
-       struct snd_soc_card *card = runtime->card;
        int num_pins, ret;
 
        jack = snd_soc_card_get_drvdata(card);
        snd_jack_set_key(jack->jack, SND_JACK_BTN_2, KEY_VOLUMEUP);
        snd_jack_set_key(jack->jack, SND_JACK_BTN_3, KEY_VOLUMEDOWN);
 
-       //snd_soc_component_set_jack(component, jack, NULL);
-       // TODO: Fix nau8825 codec to use .set_jack, like everyone else
-       nau8825_enable_jack_detect(component, jack);
+       return snd_soc_component_set_jack(asoc_rtd_to_codec(runtime, 0)->component, jack, NULL);
+}
 
-       return 0;
+static void avs_nau8825_codec_exit(struct snd_soc_pcm_runtime *rtd)
+{
+       snd_soc_component_set_jack(asoc_rtd_to_codec(rtd, 0)->component, NULL, NULL);
 }
 
 static int
        dl->id = 0;
        dl->dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF | SND_SOC_DAIFMT_CBS_CFS;
        dl->init = avs_nau8825_codec_init;
+       dl->exit = avs_nau8825_codec_exit;
        dl->be_hw_params_fixup = avs_nau8825_be_fixup;
        dl->ops = &avs_nau8825_ops;
        dl->nonatomic = 1;
        return 0;
 }
 
-static int avs_card_set_jack(struct snd_soc_card *card, struct snd_soc_jack *jack)
-{
-       struct snd_soc_component *component;
-
-       for_each_card_components(card, component)
-               snd_soc_component_set_jack(component, jack, NULL);
-       return 0;
-}
-
-static int avs_card_remove(struct snd_soc_card *card)
-{
-       return avs_card_set_jack(card, NULL);
-}
-
 static int avs_card_suspend_pre(struct snd_soc_card *card)
 {
-       return avs_card_set_jack(card, NULL);
+       struct snd_soc_dai *codec_dai = snd_soc_card_get_codec_dai(card, SKL_NUVOTON_CODEC_DAI);
+
+       return snd_soc_component_set_jack(codec_dai->component, NULL, NULL);
 }
 
 static int avs_card_resume_post(struct snd_soc_card *card)
            codec_dai->playback_widget->active)
                snd_soc_dai_set_sysclk(codec_dai, NAU8825_CLK_FLL_FS, 0, SND_SOC_CLOCK_IN);
 
-       return avs_card_set_jack(card, jack);
+       return snd_soc_component_set_jack(codec_dai->component, jack, NULL);
 }
 
 static int avs_nau8825_probe(struct platform_device *pdev)
        card->name = "avs_nau8825";
        card->dev = dev;
        card->owner = THIS_MODULE;
-       card->remove = avs_card_remove;
        card->suspend_pre = avs_card_suspend_pre;
        card->resume_post = avs_card_resume_post;
        card->dai_link = dai_link;