dev_info(&client->dev,
                 "chip found, driver version " DRV_VERSION "\n");
 
-       clientdata = kzalloc(sizeof(*clientdata), GFP_KERNEL);
+       clientdata = devm_kzalloc(&client->dev, sizeof(*clientdata),
+                               GFP_KERNEL);
        if (!clientdata) {
                rc = -ENOMEM;
                goto exit;
        clientdata->features = id->driver_data;
        i2c_set_clientdata(client, clientdata);
 
-       rtc = rtc_device_register(client->name, &client->dev,
-                                 &m41t80_rtc_ops, THIS_MODULE);
+       rtc = devm_rtc_device_register(&client->dev, client->name,
+                                       &m41t80_rtc_ops, THIS_MODULE);
        if (IS_ERR(rtc)) {
                rc = PTR_ERR(rtc);
                rtc = NULL;
        goto exit;
 
 exit:
-       if (rtc)
-               rtc_device_unregister(rtc);
-       kfree(clientdata);
        return rc;
 }
 
 static int m41t80_remove(struct i2c_client *client)
 {
+#ifdef CONFIG_RTC_DRV_M41T80_WDT
        struct m41t80_data *clientdata = i2c_get_clientdata(client);
-       struct rtc_device *rtc = clientdata->rtc;
 
-#ifdef CONFIG_RTC_DRV_M41T80_WDT
        if (clientdata->features & M41T80_FEATURE_HT) {
                misc_deregister(&wdt_dev);
                unregister_reboot_notifier(&wdt_notifier);
        }
 #endif
-       if (rtc)
-               rtc_device_unregister(rtc);
-       kfree(clientdata);
 
        return 0;
 }