1 files changed, 2 insertions, 4 deletions

diff --git a/drivers/dax/dax.c b/drivers/dax/dax.c
index b891a129b275..803f3953b341 100644
--- a/drivers/dax/dax.c
+++ b/drivers/dax/dax.c
@@ -211,11 +211,9 @@ int devm_create_dax_dev(struct dax_region *dax_region, struct resource *res,
         }
         dax_dev->dev = dev;
 
-        rc = devm_add_action(dax_region->dev, unregister_dax_dev, dev);
-        if (rc) {
-                unregister_dax_dev(dev);
+        rc = devm_add_action_or_reset(dax_region->dev, unregister_dax_dev, dev);
+        if (rc)
                 return rc;
-        }
 
         return 0;