1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/macintosh/therm_adt746x.c b/drivers/macintosh/therm_adt746x.c
index c42eeb43042d..16d82f17ae82 100644
--- a/drivers/macintosh/therm_adt746x.c
+++ b/drivers/macintosh/therm_adt746x.c
@@ -182,6 +182,7 @@ remove_thermostat(struct i2c_client *client)
 
         thermostat = NULL;
 
+        i2c_set_clientdata(client, NULL);
         kfree(th);
 
         return 0;
@@ -399,6 +400,7 @@ static int probe_thermostat(struct i2c_client *client,
         rc = read_reg(th, CONFIG_REG);
         if (rc < 0) {
                 dev_err(&client->dev, "Thermostat failed to read config!\n");
+                i2c_set_clientdata(client, NULL);
                 kfree(th);
                 return -ENODEV;
         }