1 files changed, 3 insertions, 2 deletions

diff --git a/sound/soc/codecs/wm8997.c b/sound/soc/codecs/wm8997.c
index a4d11770630c..e9c4a9f35392 100644
--- a/sound/soc/codecs/wm8997.c
+++ b/sound/soc/codecs/wm8997.c
@@ -1055,13 +1055,14 @@ static struct snd_soc_dai_driver wm8997_dai[] = {
 
 static int wm8997_codec_probe(struct snd_soc_codec *codec)
 {
+        struct snd_soc_dapm_context *dapm = snd_soc_codec_get_dapm(codec);
         struct wm8997_priv *priv = snd_soc_codec_get_drvdata(codec);
 
         arizona_init_spk(codec);
 
-        snd_soc_dapm_disable_pin(&codec->dapm, "HAPTICS");
+        snd_soc_dapm_disable_pin(dapm, "HAPTICS");
 
-        priv->core.arizona->dapm = &codec->dapm;
+        priv->core.arizona->dapm = dapm;
 
         return 0;
 }